Vijos 1313 金明的预算方案 （分组背包）

题目链接

和多校上的黄金矿工很类似，貌似这个还简单点，每个主件，最多有2个附件，讨论下就可以，当然也神似树形DP。自己做的时候SB了，边输入边处理了，想当然了，果断悲剧了，交了N次之后，发现把背包分组的会错误，最可恨的是不知道该怎么搞了，想快排的，结构体神马的，发现从0开始，后来写的会很麻烦。。受不了了，把以前的代码改改，先把根找到，然后枚举他的孩子。。。虽然效率低点，终于DEBUG好几个小时后，终于乱搞过了。。。代码能力弱爆了，不会乱搞啊。。。

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
int v[50000],bag[201][10],que[201],c[201][10],q[201],p[201],o[201];
int main()
{
    int i,j,k,n,m,num;
    scanf("%d%d",&n,&m);
    n = n/10;
    num = 1;
    for(i = 1; i <= m; i ++)
    {
        scanf("%d%d%d",&bag[i][1],&p[i],&q[i]);
        bag[i][1] /= 10;
        if(q[i] == 0)
        {
            o[i] = 1;
            que[num++] = i;
            c[i][1] = bag[i][1]*p[i];
        }
    }
    for(i = 1; i <= m; i ++)
    {
        if(o[i] == 1)
        {
            k = 1;
            for(j = 1;j <= m;j ++)
            {
                if(q[j] == i&&k == 1)
                {
                    k ++;
                    bag[i][2] = bag[i][1] + bag[j][1];
                    c[i][2] = c[i][1] + bag[j][1]*p[j];
                }
                else if(q[j] == i&&k == 2)
                {
                    bag[i][3] = bag[i][1] + bag[j][1];
                    c[i][3] = bag[i][1] + bag[j][1]*p[j];
                    bag[i][4] = bag[i][2] + bag[j][1];
                    c[i][4] = c[i][2] + bag[j][1]*p[j];
                }
            }
        }
    }
    for(i = 1; i <= num-1; i ++)
    {
        for(j = n; j >= bag[que[i]][1]; j --)
        {
            for(k = 1; k <= 4; k ++)
            {
                if(j-bag[que[i]][k] >= 0&&v[j] < v[j-bag[que[i]][k]]+c[que[i]][k])
                    v[j] = v[j-bag[que[i]][k]]+c[que[i]][k];
            }
        }
    }
    printf("%d\n",v[n]*10);
    return 0;
}