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;
}


  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值