POJ 2468 Apple Tree 樹上瞎搞分組背包

標籤：style   blog   class   code   ext   int   

昨晚Debug了好久始終找不出哪裡錯了，今早再一看發現自己已榮升逗比Beta 2.0 Version.

個人感覺此題為HDU 4003 的弱化版。

把每棵子樹都看成一類商品，在每類商品中至多選一件。則問題轉化為最基本的分組背包問題。

dp[s][c][k] c == 1時，表示在s結點不返回時走K的最大收益，c == 0時，表示在s結點重新返回時走k步的最大收益。

可以dfs從底到頂更新dp。值得一提的是，要保證每個結點的dp[s][c][k]只會被dfs一次，即不能重複更新。

因為重複的dfs不能保證每類商品中之多選一件。

#include <iostream>#include <algorithm>#include <cstdlib>#include <cstdio>#include <cstring>#include <queue>#include <cmath>#include <stack>#include <map>#pragma comment(linker, "/STACK:1024000000");#define EPS (1e-8)#define LL long long#define ULL unsigned long long LL#define _LL __int64#define _INF 0x3f3f3f3f#define Mod 1000000007#define LM(a,b) (((ULL)(a))<<(b))#define RM(a,b) (((ULL)(a))>>(b))using namespace std;const LL MAXN = 10010;struct N{    int u,v,w,next;} edge[MAXN*2];int head[MAXN];int Top;void Link(int u,int v,int w = -1){    edge[Top].u = u;    edge[Top].v = v;    edge[Top].w = w;    edge[Top].next = head[u];    head[u] = Top++;}void Init_head_Top(int n){    memset(head,-1,sizeof(int)*(n+2));    Top = 0;}LL dp[110][2][210];LL price[210];void dfs(int s,int pre,int c,int k){    if(k < 0)        return ;    if(dp[s][c][k] != -1)        return ;    dp[s][1][0] = price[s];    dp[s][0][0] = price[s];    for(int p = head[s] ; p != -1; p = edge[p].next)    {        if(edge[p].v != pre)        {            if(c == 1)                dfs(edge[p].v,s,1,k-1);            dfs(edge[p].v,s,0,k-2);        }    }    for(int p = head[s] ; p != -1; p = edge[p].next)    {        if(edge[p].v != pre)        {            if(c == 1)            {                for(int i = k;i >= 0; --i)                {                    if(dp[s][1][i] != -1)                    {                        for(int j = k-i-2;j >= 0; --j)                        if(dp[edge[p].v][0][j] != -1)                            dp[s][1][i+j+2] = max(dp[s][1][i+j+2],dp[s][1][i] + dp[edge[p].v][0][j]);                    }                    if(dp[s][0][i] != -1)                    {                        for(int j = k-i-1;j >= 0 ; --j)                        if(dp[edge[p].v][1][j] != -1)                            dp[s][1][i+j+1] = max(dp[s][1][i+j+1],dp[s][0][i] + dp[edge[p].v][1][j]);                    }                }            }            for(int i = k;i >= 0; --i)            {                if(dp[s][0][i] != -1)                {                    for(int j = k-i-2;j >= 0; --j)                    if(dp[edge[p].v][0][j] != -1)                        dp[s][0][i+j+2] = max(dp[s][0][i+j+2],dp[s][0][i] + dp[edge[p].v][0][j]);                }            }        }    }    if(c == 1)    {        for(int i = 0;i <= k; ++i)            dp[s][c][i] = max(dp[s][c][i],(LL)0);    }    for(int i = 0;i <= k; ++i)        dp[s][0][i] = max(dp[s][0][i],(LL)0);}int main(){    //freopen("ts.txt","w",stdout);    int n,k,i,u,v;    while(scanf("%d %d",&n,&k) != EOF)    {        memset(dp,-1,sizeof(dp));        Init_head_Top(n);        for(i = 1; i <= n; ++i)            scanf("%lld",&price[i]);        for(i = 1; i < n; ++i)        {            scanf("%d %d",&u,&v);            Link(u,v);            Link(v,u);        }        dfs(1,-1,1,k);        LL Max = 0;//        int ww = k;//        for(i = 1;i <= n; ++i)//        {//            for(int j = 0;j <= 1; ++j)//            {//                for(k = 0;k <= ww; ++k)//                {//                    if(dp[i][j][k] > 0)//                    printf("i = %d j = %d k = %d dp = %lld\n",i,j,k,dp[i][j][k]);//                }//            }//        }////        k = ww;        for(i = 0; i <= k; ++i)        {            Max = max(Max,dp[1][1][i]);        }        printf("%lld\n",Max);    }    return 0;}