牛客网暑期ACM多校训练营（第三场）A 多维DP + 方案状态压缩

题目链接

---

题意：
给定$n$个物品，每个物品有四维的$cost$和一个$val$，选择若干个物品，每个物品只能选取一次，问在不超过给出的四维限制的情况下，能够达到的最大的$val$是多少？输出选择方案($spj$)
所有参数均$<=36$

---

思路：
01背包或者折半枚举均能做，但主要难处理的地方在于如何输出方案。
如果使用传统的记录方式，需要使用一个$O(n^5)$的数组来回溯，对于内存消耗来说相当危险。
因为$n<=36$，考虑使用一个$long$ $long$值，将每个选择方案状态压缩，一个二进制位代表一个物品的选择情况。
则状态转移时，不仅仅转移最大的$val$值，同时转移方案，对于前一个状态的选择方案$preSta$，通过选择第$i$个物品转移到当前状态$nowSta$，则：
$nowSta=preSta|(1<<i)$

这样的话时间复杂度$O(n^5)$，空间复杂度$O(n^4)$

此题得解。

---

代码：

#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;
}