【背包DP】洛谷P1060 开心的金明 题解

洛谷P1060 开心的金明 题解

题目传送门

 

分析：

又是背包问题中大名鼎鼎的金明系列，与普通的背包不同，这道题有了“主件”和“附件”的概念

但实际上我们并不需要单独考虑附件，只需要在对主件进行决策的时候同时考虑取附件的情况。

那么对于一个主件来说，无非有四种情况：

1. 只取主件，
2. 取主件和一个附件(如果存在)：，这里f1是a[i]的一个附件
3. 取主件和第二个附件(如果存在)：，f2是第二个附件
4. 取主件和两个附件：

当然，以上条件均要满足f下标不为负数，所以要

至此对于一个主件，就全部考虑完了，那么它的附件也就不用考虑了。

代码

#include <bits/stdc++.h>

using namespace std;

struct node {
    int w, v, q;
    int f1, f2;

    node() { f1 = f2 = 0; }
} a[65];

int f[32005];

void put(int i, int j) {     //把i作为j的附件
    if (a[j].f1) {
        a[j].f2 = i;
    } else {
        a[j].f1 = i;
    }
}

int main() {
    int m, n;
    cin >> m >> n;
    for (int i = 1; i <= n; ++i) {
        cin >> a[i].v >> a[i].w >> a[i].q;
        a[i].w *= a[i].v;
        if (a[i].q > 0) {
            put(i, a[i].q);
        }
    }
    bool flg1, flg2;
    for (int i = 1; i <= n; ++i) {
        if (a[i].q)
            continue;
        flg1 = flg2 = false;
        for (int j = m; j >= a[i].v; --j) {
            f[j] = max(f[j], f[j - a[i].v] + a[i].w);
            int f1 = a[i].f1;
            int f2 = a[i].f2;
            if (f1 && j >= (a[i].v + a[f1].v)) {
                f[j] = max(f[j], f[j - a[i].v - a[f1].v] + a[i].w + a[f1].w);
            }
            if (f2 && j >= (a[i].v + a[f2].v)) {
                f[j] = max(f[j], f[j - a[i].v - a[f2].v] + a[i].w + a[f2].w);
            }
            if (f2 && j >= (a[i].v + a[f2].v + a[f1].v)) {
                f[j] = max(f[j], f[j - (a[i].v + a[f2].v + a[f1].v)] + a[i].w + a[f1].w + a[f2].w);
            }
        }
    }
    cout << f[m] << endl;

    return 0;
}