https://www.luogu.org/problemnew/show/P1273
某收费有线电视网计划转播一场重要的足球比赛。他们的转播网和用户终端构成一棵树状结构,这棵树的根结点位于足球比赛的现场,树叶为各个用户终端,其他中转站为该树的内部节点。
从转播站到转播站以及从转播站到所有用户终端的信号传输费用都是已知的,一场转播的总费用等于传输信号的费用总和。
现在每个用户都准备了一笔费用想观看这场精彩的足球比赛,有线电视网有权决定给哪些用户提供信号而不给哪些用户提供信号。
写一个程序找出一个方案使得有线电视网在不亏本的情况下使观看转播的用户尽可能多。
输入输出格式
输入格式:
输入文件的第一行包含两个用空格隔开的整数N和M,其中2≤N≤3000,1≤M≤N-1,N为整个有线电视网的结点总数,M为用户终端的数量。
第一个转播站即树的根结点编号为1,其他的转播站编号为2到N-M,用户终端编号为N-M+1到N。
接下来的N-M行每行表示—个转播站的数据,第i+1行表示第i个转播站的数据,其格式如下:
K A1 C1 A2 C2 … Ak Ck
K表示该转播站下接K个结点(转播站或用户),每个结点对应一对整数A与C,A表示结点编号,C表示从当前转播站传输信号到结点A的费用。最后一行依次表示所有用户为观看比赛而准备支付的钱数。
输出格式:
输出文件仅一行,包含一个整数,表示上述问题所要求的最大用户数。
题意:在一棵树中,传给每一个结点有一个费用,传给用户的话用户会交钱,求不亏本的情况下最多接通多少用户。
思路:树上背包,设f(u,i):以u为根的子树,接通i个用户最多赚多少钱。枚举子结点来进行转移。
和这道题https://blog.csdn.net/Wen_Yongqi/article/details/87986821差不多。
初始化全部为-INF。
有一个优化(决定后四个点过不过):第二维未必要循环m次,只需最多循环【截至目前的子树有多少用户】次
#include<bits/stdc++.h>
using namespace std;
#define maxn 3000+100
int n,m,f[maxn][maxn],w[maxn],cost[maxn],sz[maxn];
vector<int> child[maxn];
void dfs(int u)
{
if(child[u].size()==0)
{
f[u][1]=w[u]-cost[u];
sz[u]=1;
return;
}
f[u][0]=-cost[u];
for(int i=0;i<child[u].size();i++)
{
int v=child[u][i];
dfs(v);
sz[u]+=sz[v];
for(int j=min(sz[u],m);j>=1;j--)
{
for(int k=1;k<=j;k++)
f[u][j]=max(f[u][j],f[u][j-k]+f[v][k]);
}
}
}
int main()
{
// freopen("input.in","r",stdin);
int k,x,y;
cin>>n>>m;
for(int i=1;i<=n-m;i++)
{
scanf("%d",&k);
while(k--)
{
scanf("%d%d",&x,&y);
child[i].push_back(x);
cost[x]=y;
}
}
for(int i=n-m+1;i<=n;i++)scanf("%d",&w[i]);
for(int i=1;i<=n;i++)for(int j=0;j<=m;j++)f[i][j]=-(1<<27);
dfs(1);
for(int i=m;i>=1;i--)if(f[1][i]>=0)
{
cout<<i<<endl;
exit(0);
}
cout<<0<<endl;
return 0;
}