洛谷p1064

https://www.luogu.org/problemnew/show/P1064

带有附件的背包问题，它属于01背包的变式。

这题还好，每一个物品最多只有两个附件，那么我们在对主件进行背包的时候，决策就不再是两个了，而是五个。

还记得01背包的决策是什么吗？

1.不选，然后去考虑下一个

2.选，背包容量减掉那个重量，总值加上那个价值。

这个题的决策是五个，分别是：

1.不选，然后去考虑下一个

2.选且只选这个主件

3.选这个主件，并且选附件1

4.选这个主件，并且选附件2

5.选这个主件，并且选附件1和附件2.

这个。。。很好想吧。。。

我们知道，01背包的状态转移方程（已使用滚动数组优化）是f[j] = max(f[j],f[j-w[i]]+c[i])，那么，这道题的转移方程也就不难写出了。

等等，你得先判断某个选附件的决策是不是可行的，如果当前的容量还够放第一个，或第二个，或两个都选的附件，那么才能考虑转移。

当然，不选附件的话就不用判啦，直接01背包的转移方程即可。

我们令main_item_w数组表示某个主件的费用，而main_item_c数组表示某个主件的价值。

同样的，用二维数组annex_item_w表示某个附件的费用，annex_item_c表示某个附件的价值，第二维只需要0,1,2这三个数，其中第二维是0的场合表示这个主件i的附件数量，它只能等于0或1或2。第二维是1或者是2的值代表以i为主件的附件1或者附件2的相关信息（费用价值）。这些数组的信息应该在读入时处理好，具体详见代码。

这样，状态转移方程就是四个。

不选附件的①：f[j] = max(f[j],f[j-main_item_w[i]]+main_item_c[i]);

选附件1的②：f[j] = max(f[j],f[ j - main_item_w[i] - annex_item_w[i][1] ] + main_item_c[i] + annex_item_c[i][1]);

选附件2的③：f[j] = max(f[j],f[ j - main_item_w[i] - annex_item_w[i][2] ] + main_item_c[i] + annex_item_c[i][2]);

选附件1和附件2的④：f[j] = max(f[j],f[ j - main_item_w[i] - annex_item_w[i][1] - annex_item_w[i][2] ] + main_item_c[i] + annex_item_c[i][1] + annex_item_c[i][2]);

已经滚动掉了第一维，道理和正常向的01背包都是一样的，即只有i和i-1有关系，但是这个规律在循环中已经满足了所以完全没必要记录。

目标状态f[n]，输出就好。

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cmath>
 4 #include<algorithm>
 5 using namespace std;
 6 typedef long long ll;
 7 int f[30010];
 8 int n, m;
 9 int v, p, q;
10 int ma[30010];
11 int mac[30010];
12 int an[30010][3];
13 int anc[30010][3];
14 int main()
15 {
16     cin >> n >> m;
17     for (int i = 1; i <= m; i++)
18     {
19         cin >> v >> p >> q;
20         if (q == 0)
21         {
22             ma[i] = v;
23             mac[i] = v * p;
24         }
25         else
26         {
27             an[q][0]++;
28             an[q][an[q][0]] = v;
29             anc[q][an[q][0]] = v * p;
30         }
31     }
32     for (int i = 1; i <= m; i++)
33     {
34         for (int j = n; j >= ma[i] && ma[i] != 0; j--)
35         {
36             f[j] = max(f[j], f[j - ma[i]] + mac[i]);
37             if (j >= ma[i] + an[i][1])
38             {
39                 f[j] = max(f[j], f[j - ma[i] - an[i][1]] + mac[i] + anc[i][1]);
40             }
41             if (j >= ma[i] + an[i][2])
42             {
43                 f[j] = max(f[j], f[j - ma[i] - an[i][2]] + mac[i] + anc[i][2]);
44             }
45             if (j >= ma[i] + an[i][1] + an[i][2])
46             {
47                 f[j] = max(f[j], f[j - ma[i] - an[i][1] - an[i][2]] + mac[i] + anc[i][1] + anc[i][2]);
48             }
49         }
50     }
51     cout << f[n] << endl;
52     return 0;
53 }

猜你喜欢