用贪心算法写出01背包问题代码
时间: 2023-07-12 08:27:43 浏览: 43
好的,下面是用贪心算法解决01背包问题的代码实现:
```
#include <iostream>
#include <algorithm>
using namespace std;
// 物品结构体
struct Item {
int weight; // 物品重量
int value; // 物品价值
};
// 按照单位重量的价值从大到小排序
bool cmp(Item a, Item b) {
return (double)a.value / a.weight > (double)b.value / b.weight;
}
// 01背包问题贪心算法
double knapsack_greedy(Item arr[], int n, int W) {
sort(arr, arr + n, cmp); // 按照单位重量的价值从大到小排序
double total_value = 0; // 总价值
for (int i = 0; i < n; i++) {
if (arr[i].weight <= W) { // 若物品可以全部放入背包
total_value += arr[i].value;
W -= arr[i].weight;
}
else { // 若物品放不完,则将部分物品放入背包中
total_value += (double)arr[i].value * W / arr[i].weight;
break;
}
}
return total_value;
}
// 测试代码
int main() {
int W = 50; // 背包容量
Item items[] = {{10, 60}, {20, 100}, {30, 120}}; // 物品重量和价值
int n = sizeof(items) / sizeof(items[0]); // 物品数量
double max_value = knapsack_greedy(items, n, W); // 计算最大价值
cout << "最大价值为:" << max_value << endl;
return 0;
}
```
以上代码中,我们先定义了一个物品结构体,包含物品的重量和价值,然后实现了一个按照单位重量的价值从大到小排序的比较函数,最后实现了一个贪心算法的函数,计算出最大价值。最后在 main 函数中进行测试,输入背包容量和物品的重量和价值,输出最大价值。