解决整数背包

时间:2012-06-14 15:44:29

标签: c++ algorithm dynamic-programming knapsack-problem

我是一个新的动态编程,并在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

2 个答案:

答案 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打印结果,这是荒谬的。