背包问题 变式 DP

如果单独将某个物品去除，每次做一个01背包。
时间复杂度为O(nqnv)

如果将其拆分为，两个体积为V1、V2的背包。
采用前缀背包和后缀背包。拆成两部分

#include <stdio.h>
#include <utility>
#include <math.h>
#include <algorithm>
#include <string.h>
#include <vector>
using namespace std;
// ================= 代码实现开始 =================

const int N = 5005;

//d:前缀背包 d[i][j]表示物品1到物品i放进容量为j的背包中的最大价值
//f:后缀背包 f[i][j]表示物品i到物品n放进容量为j的背包中的最大价值
int d[N][N]={0},f[N][N]={0};

// n个物品，每个物品有体积价值，求若扔掉一个物品后装进给定容量的背包的最大价值
// n：如题
// w：长度为n+1的数组，w[i]表示第i个物品的价值（下标从1开始，下标0是一个数字-1，下面同理）
// v：长度为n+1的数组，v[i]表示第i个物品的体积
// q：如题
// qV：长度为q+1的数组，qV[i]表示第i次询问所给出的背包体积
// qx：长度为q+1的数组，qx[i]表示第i次询问所给出的物品编号
// 返回值：返回一个长度为q的数组，依次代表相应询问的答案
vector<int> getAnswer(int n, vector<int> w, vector<int> v, int q, vector<int> qV, vector<int> qx) {
    //计算前缀背包
    for(int i=1; i<=n; i++){
        for(int V=0; V<v[i]; ++V)
            d[i][V] = d[i-1][V];
        for(int V=v[i]; V<=5000; ++V)
            d[i][V] = max(d[i-1][V], d[i-1][V-v[i]]+w[i] );
    }
    //计算后缀背包，计算过程与前缀背包类似
    for(int i=n; i>=1; --i){
        for(int V=0; V<v[i]; ++V)
            f[i][V] = f[i+1][V];
        for(int V=v[i]; V<=5000; ++V)
            f[i][V] = max(f[i+1][V], f[i+1][V-v[i]]+w[i] );
    }
    
/*  for(int i=1; i<=n; i++){
        for(int V=0; V<7; ++V)
        printf("%d,",d[i][V]);
        printf("\n");
    }
    for(int i=1; i<=n; i++){
        for(int V=0; V<7; ++V)
        printf("%d,",f[i][V]);
        printf("\n");
    }
 */

    vector<int> ans;
    for(int k=1; k<=q; ++k){
        int x=qx[k],V = qV[k];
        int mx = 0; //记录当前询问的最优答案
        //抽去x，然后用前缀背包与后缀背包计算最后背包
        for(int i=0; i<=V; i++){
            mx = max(mx,d[x-1][i]+f[x+1][V-i]);
//            printf("x=%d i=%d,%d,%d\n",x,i,d[x-1][i],f[x+1][V-i]);
        }
// 拆为体积i和V-i的背包。
        ans.push_back(mx);
    }
    return ans;
} 

// ================= 代码实现结束 =================

int main() {
    int n, q;
    vector<int> v, w, qv, qx;
    v.push_back(-1);
    w.push_back(-1);
    qv.push_back(-1);
    qx.push_back(-1);
    scanf("%d", &n);
    for (int i = 0; i < n; ++i) {
        int a, b;
        scanf("%d%d", &a, &b);
        v.push_back(a);
        w.push_back(b);
    }
    scanf("%d", &q);
    for (int i = 0; i < q; ++i) {
        int a, b;
        scanf("%d%d", &a, &b);
        qv.push_back(a);
        qx.push_back(b);
    }
    vector<int> ans = getAnswer(n, w, v, q, qv, qx);
    for (int i = 0; i < q; ++i)
        printf("%d\n", ans[i]);
    return 0;
}