Codevs1155 金明的预算方案 ——2006年NOIP全国联赛提高组 变种经典背包dp

Codevs1155 金明的预算方案 ——2006年NOIP全国联赛提高组

由于至多有2个附件，直接手动枚举就好了。
dp时：

• 如果当前物品是附件， 跳过
• 如果当前物品时主件
  
  • 只选这个主件，不选它的附件
  • 当这个主件有附件时， 选它和它的第一个主件
  • 当这个主件有2个附件时， 选它和它的第二个主件
  • 当这个主件有2个附件时， 选它和它的所有附件

#include <cstdio>#include <cstring>#include <iostream>#include <cstdlib>#include <vector>using namespace std;#define MAXN (60+6)#define MAXV (32000+32)struct item{    int v, p, q;}l[MAXN];int dp[MAXV];vector <int> q[MAXN];int main(){    int V, n;    cin >> V >> n;    for(int i = 1; i <= n; i ++)    {        scanf("%d%d%d", &l[i].v, &l[i].p, &l[i].q);        if(l[i].q) q[l[i].q].push_back(i);    }    for(int i = 1; i <= n; i ++)    {        if(l[i].q) continue;        for(int j = V; j >= 0; j --)        {            // 只选主件            if(j >= l[i].v)                dp[j] = max(dp[j], dp[j - l[i].v] + l[i].v*l[i].p);            //只选第一件             if(q[i].size() && j >= l[i].v + l[q[i][0]].v)                dp[j] = max(dp[j], dp[j - (l[i].v + l[q[i][0]].v)] + l[i].v*l[i].p + l[q[i][0]].v*l[q[i][0]].p);            //只选第二件            if(q[i].size() > 1 && j >= l[i].v + l[q[i][1]].v)                dp[j] = max(dp[j], dp[j - (l[i].v + l[q[i][1]].v)] + l[i].v*l[i].p + l[q[i][1]].v*l[q[i][1]].p);            //全选            if(q[i].size() > 1 && j >= l[i].v + l[q[i][0]].v + l[q[i][1]].v)                dp[j] = max(dp[j], dp[j - (l[i].v + l[q[i][0]].v + l[q[i][1]].v)] + l[i].v*l[i].p + l[q[i][0]].v*l[q[i][0]].p + l[q[i][1]].v*l[q[i][1]].p);        }    }    cout << dp[V];    return 0;} 

另:(精简不了多少……)

#include <cstdio>#include <cstring>#include <iostream>#include <cstdlib>#include <vector>using namespace std;#define MAXN (60+6)#define MAXV (32000+32)int dp[MAXV];struct item{    int v, p, q;}l[MAXN];vector <int> q[MAXN];int main(){    int V, n;    cin >> V >> n;    for(int i = 1; i <= n; i ++)    {        scanf("%d%d%d", &l[i].v, &l[i].p, &l[i].q);        if(l[i].q) q[l[i].q].push_back(i);    }    for(int i = 1; i <= n; i ++)    {        if(l[i].q) continue; //如果是附件        for(int j = V; j >= 0; j --)        {            int lson, rson;            if(q[i].size()) lson = q[i][0];// 假设附件为 儿子             if(q[i].size() > 1) rson = q[i][1];            //只选主件            if(j >= l[i].v)                dp[j] = max(dp[j], dp[j-l[i].v] + l[i].v*l[i].p);            //选主件和第一个附件            if(q[i].size() > 0 && j >= l[i].v + l[lson].v)                dp[j] = max(dp[j], dp[j - (l[i].v + l[lson].v)] + l[i].v*l[i].p + l[lson].v*l[lson].p);            //选主件和第二个附件            if(q[i].size() > 1 && j >= l[i].v + l[rson].v)                dp[j] = max(dp[j], dp[j - (l[i].v + l[rson].v)] + l[i].v*l[i].p + l[rson].v*l[rson].p);            //全选            if(q[i].size() > 1 && j >= l[i].v + l[lson].v + l[rson].v)                dp[j] = max(dp[j], dp[j - (l[i].v + l[lson].v + l[rson].v)] + l[i].v*l[i].p + l[lson].v*l[lson].p + l[rson].v*l[rson].p);        }     }    cout << dp[V];    return 0;}