POJ 1155 經典樹形dp+分組揹包 + 模板

題意:電視臺發送信號給很多用戶,每個用戶有願意出的錢,電視臺經過的路線都有一定費用,求電視臺不損失的情況下最多給多少用戶發送信號。
dp[i][j]代表i節點爲根節點的子樹j個用戶的時候最大剩餘費用。
dp[i][j] = max(dp[i][j], dp[i][k]+dp[son][j-k]-w[i][son]);
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 3e3+7;
int n,m,num,head[maxn],c[maxn],dp[maxn][maxn];
struct list1{int u,v,w,next;}a[maxn];
void add(int u,int v,int w){a[num]={u,v,w,head[u]};head[u]=num++;}
void dfs(int u)
{
    int i,j,k,v,w;
    for(i=head[u];~i;i=a[i].next)
    {
        v=a[i].v;w=a[i].w;
        dfs(v);c[u]+=c[v];
        for(j=c[u];j>0;j--)              //注意循環從後往前,防止相互影響(類似於揹包)
            for(k=1;k<=j;k++)
                dp[u][j]=max(dp[u][j],dp[v][k]+dp[u][j-k]-w);
    }
}
int main()
{
    int i,j,v,w,x;
    while(~scanf("%d%d",&n,&m))
    {
        for(i=1;i<=n;i++)
            for(j=1;j<=m;j++)
                if(!j) dp[i][j]=0;
                else dp[i][j]=-1000000;
        memset(head,-1,sizeof(head));
        memset(c,0,sizeof(c));num=1;
        for(i=1;i<=n-m;i++)
        {
            scanf("%d",&x);
            for(j=1;j<=x;j++) scanf("%d%d",&v,&w),add(i,v,w);
        }
        for(i=n-m+1;i<=n;i++) scanf("%d",&dp[i][1]),c[i]=1;dfs(1);
        for(i=m;i>=0;i--) if(dp[1][i]>=0) break;printf("%d\n",i);
    }
    return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章