HDU 3236 Gift Hunting 01背包(*3) 有物品必须取

题意: 有两张购物卡, 面额为v1, v2, 有一次一键购买机会(不花钱), 物品有价值和花费, 一些物品是必须取的, 问获得的最大价值是多少.

思路:

①买的物品分到两张购物卡和一次一键购买上, 相当于3个01背包.

②在更新dp数组时, 如果当前物品是必须取的, Ⅰ它不能由自己转移(不能不取当前物品), Ⅱ如果不能由其他状态转移, 那么这个状态就是不可达的.

代码:

#include<bits/stdc++.h>

using namespace std;

void debug_out() {
    cerr << '\n';
}

template<typename T, typename ...R>
void debug_out(const T &f, const R &...r) {
    cerr << f << " ";
    debug_out(r...);
}

#define debug(...) cerr << "[" << #__VA_ARGS__ << "]: ", debug_out(__VA_ARGS__);

typedef long long ll;

const int M = 2e5 + 5;
const int inf = 1e9 + 5;
const int mod = 1e9 + 7;

int v1, v2, n;
int v[305];
int w[305];
int s[305];

int dp[505][55][2];


void init() {
    for (int i = 0; i < n; i++) {
        scanf("%d%d%d", &v[i], &w[i], &s[i]);
    }
    memset(dp, 0, sizeof(dp));
}

void solve() {
    for (int i = 0; i < n; i++) {
        for (int j = v1; j >= 0; --j) {
            for (int k = v2; k >= 0; --k) {
            /*如果i物品必须取*/
                if (s[i] == 1) {

                /*买完i后没有一件购买权了*/
                    //之所以先修改dp[j][k][0], 而不是dp[j][k][1]
                    //是因为dp[j][k][0]要用到老的(没有在此i中修改过的)dp[j][k][1].
                    int tmp = -inf;//因为i物品不能不取,所以tmp的初始值不能是dp[j][k][0].
                    int flag = 0;
                    if (j >= v[i])//①用1号购物卡买
                        if (tmp < dp[j - v[i]][k][0] + w[i]) {
                            tmp = dp[j - v[i]][k][0] + w[i];
                            flag = 1;
                        }
                    if (k >= v[i])//②用2号购物卡买
                        if (tmp < dp[j][k - v[i]][0] + w[i]) {
                            tmp = dp[j][k - v[i]][0] + w[i];
                            flag = 1;
                        }
                    if (tmp < dp[j][k][1] + w[i]) {
                        //③因为买完i后(dp[j][k][0])没有一键购买, 所以可能是这次用掉了
                        tmp = dp[j][k][1] + w[i];
                        flag = 1;
                    }

                    //因为当前的物品是必须取的
                    //所以这个物品如果不能放进去(走过3个if后flag还是0)
                    //这个情况的dp值就要赋为-inf, 代表这个情况不合法
                    if (flag == 0)
                        dp[j][k][0] = -inf;
                    else
                        dp[j][k][0] = tmp;

                /*买完i后还有一键购买权*/
                    flag = 0;
                    tmp = 0;
                    if (j >= v[i])//①
                        if (tmp < dp[j - v[i]][k][1] + w[i]) {
                            tmp = dp[j - v[i]][k][1] + w[i];
                            flag = 1;
                        }
                    if (k >= v[i])//②
                        if (tmp < dp[j][k - v[i]][1] + w[i]) {
                            tmp = dp[j][k - v[i]][1] + w[i];
                            flag = 1;
                        }

                    if (flag == 0)
                        dp[j][k][1] = -inf;
                    else
                        dp[j][k][1] = tmp;


                }
            /*如果i物品不必须取*/
                else {

                    /*买完i后没有一键购买权*/
                    int tmp = dp[j][k][0];//因为i物品可以不取
                                        // 所以tmp的初始值是dp[j][k][0](代表不取当前的物品).
                    if (tmp < dp[j][k][1] + w[i]) {//③
                        tmp = dp[j][k][1] + w[i];
                    }
                    if (j >= v[i])//①
                        if (tmp < dp[j - v[i]][k][0] + w[i]) {
                            tmp = dp[j - v[i]][k][0] + w[i];
                        }
                    if (k >= v[i])//②
                        if (tmp < dp[j][k - v[i]][0] + w[i]) {
                            tmp = dp[j][k - v[i]][0] + w[i];
                        }
                    dp[j][k][0] = tmp;

                /*买完i后还有一键购买权*/
                    tmp = dp[j][k][1];//因为i物品可以不取
                                    // 所以tmp的初始值是dp[j][k][1](代表不取当前的物品).
                    if (j >= v[i])//①
                        if (tmp < dp[j - v[i]][k][1] + w[i]) {
                            tmp = dp[j - v[i]][k][1] + w[i];
                        }
                    if (k >= v[i])//②
                        if (tmp < dp[j][k - v[i]][1] + w[i]) {
                            tmp = dp[j][k - v[i]][1] + w[i];
                        }
                    dp[j][k][1] = tmp;


                }
            }
        }
    }
    int ans = -1;
    for (int i = 0; i <= v1; i++) {
        for (int j = 0; j <= v2; j++) {
            if (ans < dp[i][j][0]) {
                ans = dp[i][j][0];
            }
        }
    }
    printf("%d\n", ans);
    puts("");

}

int main() {
    int caz = 0;
    while (scanf("%d%d%d", &v1, &v2, &n) && (v1 || v2 || n)) {
        init();
        printf("Case %d: ", ++caz);
        solve();
    }
    return 0;
}