[POJ1155]TELE(树形dp)

题目:

我是超链接

题意:

给定一棵树,1为根结点表示电视台,有m个叶子节点表示客户,
有n-m-1个中间节点表示中转站,每条树边有权值。
现在要在电视台播放一场比赛,每个客户愿意花费cost[i]的钱观看,
而从电视台到每个客户也都有个费用,并且经过一条边只会产生一个费用。
问电视台不亏损的情况最多有几个客户可以看到比赛

题解:

选课差不多,只不过多了路上的权值而已。其中f[i][j]表示以i为根的子树中,选j个叶子节点的最大收益
为什么j的循环要倒置呢?因为你只能用旧值更新新值,不能用新值更新新值,不然就会出现一点多选的完全背包情况

代码:

#include <cstdio>
#include <cstring>
#include <iostream>
#define N 3005
using namespace std;
int tot,nxt[N*2],point[N],v[N*2],c[N*2],size[N];
int f[N][N];
void addline(int x,int y,int z)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
    ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=z;
}
void dfs(int x,int fa)
{
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa)
      {
        dfs(v[i],x);
        size[x]+=size[v[i]];    
      }
}
void answer(int x,int fa)
{
    int j,k;
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa)
      {
        answer(v[i],x);
        for (j=size[x];j>=0;j--)
          for (k=1;k<=size[v[i]];k++)
            f[x][j]=max(f[x][j],f[x][j-k]+f[v[i]][k]-c[i]);
      }
}
int main()
{
    int n,m,i,j;
    scanf("%d%d",&n,&m);
    for (i=1;i<=n-m;i++)
    {
        int id;
        scanf("%d",&id);
        for (j=1;j<=id;j++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            addline(i,x,y);
        }
    }
    memset(f,128,sizeof(f));
    for (i=1;i<=n;i++) f[i][0]=0;
    for (i=m;i>=1;i--) scanf("%d",&f[n-i+1][1]),size[n-i+1]=1;
    dfs(1,0); 
    answer(1,0);
    for (i=m;i>=0;i--) if (f[1][i]>=0) {printf("%d",i);return 0;}
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值