HRBUST 1377 金明的预算方案

题目链接：HRBUST 1377 金明的预算方案

题目大意：

题解：
将主件和其附件分为一组，组内包含主件、主件和附件(1)、主件和附件(2)、主件和附件(1,2)至多四个物品。
对所有组进行(01)背包，每组只能选一个。

#include <cstring>
#include <iostream>
using namespace std;

int n, m, w[65][4], v[65][4], dp[32010];

int main() {
    while (cin >> n >> m) {
        memset(dp, 0, sizeof(dp));
        memset(w, 0, sizeof(w));
        memset(v, 0, sizeof(v));
        for (int i = 1, a, b, c; i <= m; ++i) {
            cin >> a >> b >> c;
            if (!c) {  // 是主件
                w[i][0] = a * b;
                v[i][0] = a;
            } else {  // 是附件
                if (!w[c][1]) {
                    w[c][1] = a * b;
                    v[c][1] = a;
                } else {
                    w[c][2] = a * b;
                    v[c][2] = a;
                }
            }
        }
        for (int i = 1; i <= m; ++i) {
            for (int j = n; j >= v[i][0]; --j) {
                dp[j] = max(dp[j], dp[j - v[i][0]] + w[i][0]);  // 只买主件
                // 如果没有相应附件则附件值为0，不影响
                if (v[i][0] + v[i][1] + v[i][2] <= j) {  // 买主件和两个附件
                    dp[j] = max(dp[j], dp[j - v[i][1] - v[i][2] - v[i][0]] + w[i][1] + w[i][2] + w[i][0]);
                }
                if (v[i][0] + v[i][1] <= j) {  // 买主件和一种附件
                    dp[j] = max(dp[j], dp[j - v[i][1] - v[i][0]] + w[i][1] + w[i][0]);
                }
                if (v[i][0] + v[i][2] <= j) {  // 买主件和另一种附件
                    dp[j] = max(dp[j], dp[j - v[i][2] - v[i][0]] + w[i][2] + w[i][0]);
                }
            }
        }
        cout << dp[n] << endl;
    }
    return 0;
}