Tsinghua Online Judge 背包问题2

题目链接<https://dsa.cs.tsinghua.edu.cn/oj/problem.shtml?id=2147>

背包问题2

---

描述

n个物品，每个物品有一个体积v和价值w。现在你要回答，把一个物品丢弃后，剩下的物品装进一个大小为V的背包里能得到的最大价值是多少。

输入

输入的第一行包含一个正整数n（n ≤ 5000）。

接下来n行，每行包含两个正整数v和w（v,w ≤ 5000），分别表示一个物品的体积和价值。

接下来一行包含一个正整数q（q ≤ 5000）,表示询问个数。

接下来q行，每行包含两个正整数V和x（V ≤ 5000，x ≤ n），表示询问将物品x丢弃以后剩下的物品装进一个大小为V的背包能得到的最大价值。

输出

输出q行，每行包含一个整数，表示询问的答案。

样例1输入

3
3 5
2 2
1 2
3
3 1
3 2
3 3

样例1输出

4
5
5

样例1解释

有3个物品，第一个物品的体积为3、价值为5，第二个物品体积为2、价值为2，第三个物品体积为1、价值为2。

有3个询问：

第一个询问是问去掉1物品后剩下的2、3物品填进一个大小为3的背包能得到的最大价值。显然2、3物品都是可以放进背包的，所以最大价值为2+2=4。

第二个询问是问去掉2物品后剩下的1、3物品填进一个大小为3的背包能得到的最大价值。若我们填3物品，我们只能得到价值2；若我们填1物品，则可以得到价值5。所以最大价值为5。

第三个询问我们同样也是填1物品，最大价值为5。

样例2

请查看下发文件内的sample2_input.txt和sample2_output.txt。

限制

对于30%的数据，n,q,v,V,w ≤ 10；

对于50%的数据，n,q,v,V,w ≤ 200。

时间：2 sec

空间：512 MB

题解：

我们可以预处理“前缀背包”、“后缀背包”，然后询问时做“背包合并”的操作。

前缀背包即指前i个物品的背包，后缀背包即指后j个物品的背包。枚举两个背包的容量相加就是背包合并操作。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string.h>
#include <stack>
using namespace std;
typedef long long LL;
struct Node{
    int v,w;
}a[5005];
int dpl[5005][5005];
int dpr[5005][5005];
int main()
{
    int n,m,x,v;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d%d",&a[i].w,&a[i].v);
    for(int i=1;i<=n;i++){
        for(int j=5000;j>=0;j--){
            if(a[i].w<=j) dpl[i][j]=max(dpl[i-1][j],dpl[i-1][j-a[i].w]+a[i].v);
            else dpl[i][j]=dpl[i-1][j];
        }
    }
    for(int i=n;i>=1;i--){
        for(int j=5000;j>=0;j--){
            if(a[i].w<=j) dpr[i][j]=max(dpr[i+1][j],dpr[i+1][j-a[i].w]+a[i].v);
            else dpr[i][j]=dpr[i+1][j];
        }
    }
    scanf("%d",&m);
    while(m--){
        scanf("%d%d",&v,&x);
        int ma=0;
        for(int i=0;i<=v;i++){
            ma=max(ma,dpl[x-1][i]+dpr[x+1][v-i]);
        }
        printf("%d\n",ma);
    }
}