洛谷 P1757 通天之分组背包

题目传送门

解题思路:

分组背包,代码有注释.

AC代码:

#include<iostream>
#include<cstdio>
#include<set>

using namespace std;

int n,m,c[1001],w[1001],f[1001],z[1001],p[1001],s[1001][1001];

int main() {
    scanf("%d%d",&m,&n);
    for(int i = 1;i <= n; i++) {
        scanf("%d%d%d",&c[i],&w[i],&z[i]);
        p[z[i]]++;
        s[z[i]][p[z[i]]] = i;
    }
    f[0] = 0;
    for(int o = 1;o <= n; o++)//组数 
        for(int j = m;j >= 1; j--)//背包容量 
            for(int i = p[o];i >= 1; i--) //当前组所有物品 
                if(j >= c[s[o][i]])
                    f[j] = max(f[j],f[j-c[s[o][i]]] + w[s[o][i]]);
    printf("%d",f[m]);
    return 0;
}