一个典型的有依赖的背包问题。这里每种物品最多只可能有三个附属品,我看错了题目条件,写了一个可能有n个附属品的版本。o(╯□╰)o
可以将物品按照附属关系先进行分组,对每组都来一次01背包。
#include <cstdio> #include <cstring> #include <algorithm> using namespace std; const int maxn = 100; const int maxv = 3201; struct Item { int cost,w,child,cnum[maxn],fa; }; Item itm[maxn]; int f[maxv],g[maxv]; int main() { int n,limv; scanf("%d%d",&limv,&n); for(int i = 1;i <= n;i++) { scanf("%d%d%d",&itm[i].cost,&itm[i].w,&itm[i].fa); if(itm[i].fa) itm[itm[i].fa].cnum[itm[itm[i].fa].child++] = i; itm[i].cost /= 10; } limv /= 10; for(int i = 1;i <= n;i++) if(!itm[i].fa) { memset(g,0,sizeof(g)); for(int j = 0;j < itm[i].child;j++) { Item &now = itm[itm[i].cnum[j]]; for(int v = limv;v >= now.cost;v--) { g[v] = max(g[v],g[v - now.cost] + now.cost * now.w); } } for(int v = limv;v >= itm[i].cost;v--) { for(int v1 = v - itm[i].cost; v1 >= 0;v1--) { f[v] = max(f[v],f[v - itm[i].cost - v1] + g[v1] + itm[i].cost * itm[i].w); } } } printf("%d ",f[limv] * 10); return 0; }