[POJ1155]TELE(树形dp)

来源:互联网 发布:百度软件管家电脑版 编辑:程序博客网 时间:2024/06/04 18:00

题目:

我是超链接

题意:

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

题解:

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

代码:

#include <cstdio>#include <cstring>#include <iostream>#define N 3005using 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;}}