题目:
题意:
给定一棵树,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;}
}