【洛谷P1064】[NOIP2006] 金明的预算方案

金明的预算方案

 

显然是个背包问题

把每个主件和它对应的附件放在一组，枚举每一组，有以下几种选法：

1.都不选

2.只选主件

3.一个主件+一个附件

4.一个主件+两个附件

 

于是就成了01背包。。

 

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,f[32010],v[65][4],w[65][4],pos[65]; //pos[i]记录编号为i的物品在数组v,w中的下标 
　　　　　　　　　　　　　　　　　　　　　　　　　　　 //w[i][0]表示第i个主件的附件个数

int main()
{
    scanf("%d%d",&n,&m);
    int V,P,Q,sum;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&V,&P,&Q);
        if(Q==0)
        {
            v[++sum][1]=V;
            w[sum][1]=P;
            pos[i]=sum;
        }
        else
        {
            w[pos[Q]][++w[pos[Q]][0]+1]=P;
            v[pos[Q]][w[pos[Q]][0]+1]=V;
        }
    }
    for(int i=1;i<=sum;i++)
     for(int j=n;j>=v[i][1];j--)
     {
         f[j]=max(f[j],f[j-v[i][1]]+w[i][1]*v[i][1]);
         if(w[i][0]>=1&&j-v[i][1]-v[i][2]>=0)
          f[j]=max(f[j],f[j-v[i][1]-v[i][2]]+w[i][1]*v[i][1]+w[i][2]*v[i][2]);
         if(w[i][0]==2&&j-v[i][1]-v[i][3]>=0)
         {
              f[j]=max(f[j],f[j-v[i][1]-v[i][3]]+w[i][1]*v[i][1]+w[i][3]*v[i][3]);
              if(j-v[i][1]-v[i][2]-v[i][3]>=0)
               f[j]=max(f[j],f[j-v[i][1]-v[i][2]-v[i][3]]+w[i][1]*v[i][1]+w[i][2]*v[i][2]+w[i][3]*v[i][3]);
        }
     }
    printf("%d\n",f[n]);
    return 0;
}