牛客网多校3 PACM Team(dp 多维)

题目:有n个物品,每个物品需要四种代价,分别需要ai,bi,ci,di。价值为val[i]。

1<<36 ??? 我是智障!1ll<<36啊!!!不过比赛时好像过了。。过题也很无奈

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
int dp[2][40][40][40][40],a[40],b[40],c[40],d[40],val[40];
ll path[2][40][40][40][40];///最多36个,状压记录路径
int A,B,C,D,n;
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=0;i<n;i++)
            scanf("%d%d%d%d%d",&a[i],&b[i],&c[i],&d[i],&val[i]);
        scanf("%d%d%d%d",&A,&B,&C,&D);
        memset(dp,0,sizeof dp);
        memset(path,0,sizeof path);
        for(int p=0;p<n;p++)
        {
            memcpy(dp[p&1],dp[(p+1)&1],sizeof dp[p&1]);
            memcpy(path[p&1],path[(p+1)&1],sizeof path[p&1]);
            for(int i=a[p];i<=A;i++)
            for(int j=b[p];j<=B;j++)
            for(int k=c[p];k<=C;k++)
            for(int l=d[p];l<=D;l++)
            {
                int tmp=dp[(p+1)&1][i-a[p]][j-b[p]][k-c[p]][l-d[p]]+val[p];
                if(tmp>dp[p&1][i][j][k][l])
                {
                    dp[p&1][i][j][k][l]=tmp;
                    path[p&1][i][j][k][l]=path[(p+1)&1][i-a[p]][j-b[p]][k-c[p]][l-d[p]]|(1ll<<p);
                }
            }
        }
        ll s=path[(n-1)&1][A][B][C][D];
        vector<int>ans;
        int k=0;
        while(s>0)
        {
            if(s&1)
                ans.push_back(k);
            s/=2ll;
            k++;
        }
        printf("%d\n",ans.size());
        if(!ans.empty())
            printf("%d",ans[0]);
        for(int i=1;i<ans.size();i++)
            printf(" %d",ans[i]);
        puts("");
    }
    return 0;
}

算的dp[37][37][37][37][37]数组爆内存,赛后看别人是定义了short,寻找路径的办法很好

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
short dp[37][37][37][37][37];
int a[40],b[40],c[40],d[40],val[40];
int A,B,C,D,n,vis[37];
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
            scanf("%d%d%d%d%d",&a[i],&b[i],&c[i],&d[i],&val[i]);
        scanf("%d%d%d%d",&A,&B,&C,&D);
        for(int p=1;p<=n;p++)
        {
            for(int i=0;i<=A;i++)
            for(int j=0;j<=B;j++)
            for(int k=0;k<=C;k++)
            for(int l=0;l<=D;l++)
            {
                int tmp1=-1;
                if(i>=a[p]&&j>=b[p]&&k>=c[p]&&l>=d[p])
                tmp1=dp[p-1][i-a[p]][j-b[p]][k-c[p]][l-d[p]]+val[p];
                int tmp2=dp[p-1][i][j][k][l];

                if(tmp2>tmp1)
                {
                    dp[p][i][j][k][l]=tmp2;
                }
                else
                {
                    dp[p][i][j][k][l]=tmp1;
                }
            }
        }
        vector<int>ans;
        for(int i=n;i>=1;i--)从后往前寻找看看当前最优的是不是由i这个物品转移过去的
        {
            if(a[i]>A||b[i]>B||c[i]>C||d[i]>D) continue;//加判断啊
            if(dp[i][A][B][C][D]!=dp[i-1][A][B][C][D])//是i这个转移过去的
            {
                A-=a[i],B-=b[i],C-=c[i],D-=d[i];
                ans.push_back(i-1);
            }
        }
        printf("%d\n",ans.size());
        if(!ans.empty())
            printf("%d",ans[0]);
        for(int i=1;i<ans.size();i++)
            printf(" %d",ans[i]);
        puts("");
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/dllpxfire/article/details/81226537