Bootstrap

牛客网暑期ACM多校训练营(第三场) - A PACM Team (01背包+状态压缩输出路径)

https://www.nowcoder.com/acm/contest/141/A

 

题意:

4维的01背包。每个数字都小于等于38。

所有开4维的空间是够的。

因为物品数量只有38,可以用状态压缩来记录路径。

 

你也可以开short数组来强行开五维数组。

 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <string>
using namespace std;
const int maxn = 38;
#define LL long long
int a[38],b[38],c[38],d[38],val[38];
int dp[38][38][38][38];
LL path[38][38][38][38];
 
 
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
    {
        scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
        scanf("%d",&val[i]);
    }
    int aa,bb,cc,dd;
    scanf("%d %d %d %d",&aa,&bb,&cc,&dd);
    for(int i=1; i<=n; i++)
        for(int q=aa; q>=a[i]; q--)
            for(int w=bb;w>=b[i]; w--)
                for(int e=cc; e>=c[i]; e--)
                    for(int r=dd; r>=d[i]; r--)
                    {
                        if(dp[q][w][e][r]<dp[q-a[i]][w-b[i]][e-c[i]][r-d[i]]+val[i]){
                            dp[q][w][e][r]=dp[q-a[i]][w-b[i]][e-c[i]][r-d[i]]+val[i];
                            path[q][w][e][r]=path[q-a[i]][w-b[i]][e-c[i]][r-d[i]]|(1LL<<(i-1));
                        }
                    }
    int num=0;
    for(int i=0;i<n;i++){
        if(path[aa][bb][cc][dd]&(1LL<<i)){
               num++;
        }
    }
    printf("%d\n",num);
    int p=0;
    for(int i=0;i<n;i++){
        if(path[aa][bb][cc][dd]&(1LL<<i)){
            if(p) printf(" ");
            printf("%d",i);
            p++;
        }
    }
    printf("\n");
 
 
    return 0;
}

 

;