题意:
给定
n
n
n个物品,每个物品有四维的
c
o
s
t
cost
cost和一个
v
a
l
val
val,选择若干个物品,每个物品只能选取一次,问在不超过给出的四维限制的情况下,能够达到的最大的
v
a
l
val
val是多少?输出选择方案(
s
p
j
spj
spj)
所有参数均
<
=
36
<= 36
<=36
思路:
01背包或者折半枚举均能做,但主要难处理的地方在于如何输出方案。
如果使用传统的记录方式,需要使用一个
O
(
n
5
)
O(n^5)
O(n5)的数组来回溯,对于内存消耗来说相当危险。
因为
n
<
=
36
n<=36
n<=36,考虑使用一个
l
o
n
g
long
long
l
o
n
g
long
long值,将每个选择方案状态压缩,一个二进制位代表一个物品的选择情况。
则状态转移时,不仅仅转移最大的
v
a
l
val
val值,同时转移方案,对于前一个状态的选择方案
p
r
e
S
t
a
preSta
preSta,通过选择第
i
i
i个物品转移到当前状态
n
o
w
S
t
a
nowSta
nowSta,则:
n
o
w
S
t
a
=
p
r
e
S
t
a
∣
(
1
<
<
i
)
nowSta = preSta | (1<<i)
nowSta=preSta∣(1<<i)
这样的话时间复杂度 O ( n 5 ) O(n^5) O(n5),空间复杂度 O ( n 4 ) O(n^4) O(n4)
此题得解。
代码:
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int A = 38;
class P{
public:
int v;
ll p;
}dp[A][A][A][A];
int p, a, c, m, g, n;
int main(){
scanf("%d",&n);
for (int i = 0; i < n; i++) {
scanf("%d%d%d%d%d", &p, &a, &c, &m, &g);
for (int x = A - 1; x >= p; x--) for (int y = A - 1; y >= a; y--)
for (int z = A - 1; z >= c; z--) for (int k = A - 1; k >= m; k--) {
if (dp[x - p][y - a][z - c][k - m].v + g > dp[x][y][z][k].v) {
dp[x][y][z][k].v = dp[x - p][y - a][z - c][k - m].v + g;
dp[x][y][z][k].p = dp[x - p][y - a][z - c][k - m].p | (1LL << i);
}
}
}
scanf("%d%d%d%d", &p, &a, &c, &m);
ll ans = dp[p][a][c][m].p;
int cnt = 0;
for (int i = 0; i < n; i++) {
if (ans & (1LL << i)) cnt++;
}
printf("%d\n",cnt);
for (int i = 0; i < n; i++) {
if (ans & (1LL << i)) printf("%d\n",i);
}
return 0;
}