我是一个新的动态编程,并在SPOJ (http://www.spoj.pl/problems/KNAPSACK/尝试了整数背包问题。但是,对于给定的测试用例,我的解决方案是没有给出正确的输出。如果您能建议我的以下实施是否正确,我将感谢您。请注意,变量back
用于回溯,我不知道该怎么做。我希望能帮助你实现回溯。感谢。
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <string>
#include <iostream>
using namespace std;
int knapsack(int value[], int weight[], int n, int C,
vector < int >&back)
{
int *M = new int[C + 1];
int k;
int i, j, tmp, pos;
for (i = 1; i <= C; i++) {
M[i] = M[i - 1];
pos = i - 1;
for (j = 0; j < n; j++) {
k = i - weight[j];
if (k >= 0)
tmp = M[k] + value[j];
if (tmp > M[i]) {
M[i] = tmp;
}
}
back.push_back(pos);
}
int ans = M[C];
delete[]M;
return ans;
}
int main()
{
int C, N;
cin >> C >> N;
int* value = new int[N+1];
int* weight = new int[N+1];
for ( int i = 0; i <= N; i++) {
cin>>value[i]>>weight[i];
}
vector < int >back(N, 0);
cout<<knapsack(value,weight,N,C,back)<<endl;
return 0;
}
以下是正确的输入/输出测试用例:
Input:
4 5
1 8
2 4
3 0
2 5
2 3
Output:
13
但是,我得到的输出是17
。
答案 0 :(得分:8)
这是Knapsack问题的一个版本,称为0-1背包。
你在代码中犯了一些愚蠢的错误。
首先输入的第一个整数是权重,第二个是值。虽然你首先是体重,第二是体重。此外,您将n + 1个值作为输入0到N(包括0和N)。
现在在您的算法中,您正在考虑可以使用任意数量的项目副本,这不是真的,这是0-1背包。阅读此http://en.wikipedia.org/wiki/Knapsack_problem。
我提出了这个算法,我提交并被接受了。所以这应该工作正常。
int M[2000][2000];
int knapsack(int value[], int weight[], int C, int n)
{
for(int i = 1; i <= C; i++){
for(int j = 0; j <n; j++){
if(j > 0){
M[j][i] = M[j-1][i];
if (weight[j] <= i)
M[j][i] = max(M[j][i], M[j-1][i-weight[j]]+value[j]);
}
else{
M[j][i] = 0;
if(weight[j] <= i)
M[j][i] = max(M[j][i], value[j]);
}
}
// cout << M[i][n-1] << endl;
}
return M[n-1][C];
}
int main()
{
int C, N;
cin >> C >> N;
// cout << C << endl;
int* value = new int[N+1];
int* weight = new int[N+1];
for ( int i = 0; i < N; i++) {
cin>>weight[i]>>value[i];
}
// vector < int >back(N, 0);
cout<<knapsack(value,weight,C,N)<<endl;
return 0;
}
BTW不动态分配数组只需使用向量
vector <int> My_array(n);
答案 1 :(得分:3)
在Python中https://sites.google.com/site/mikescoderama/Home/0-1-knapsack-problem-in-p很好地记录了背包问题的一个版本。
编辑:没关系,我跳过了第一行输入定义C和N的部分。也就是说,您的输入示例似乎没有加载您提供的代码(它正在寻找比预期更多的一对)由于&lt; = N)。我希望循环应该是&lt; N,将0..n-1作为项目。
修复我得到的134516145打印结果,这是荒谬的。