题目链接:传送门
题目描述
某收费有线电视网计划转播一场重要的足球比赛。他们的转播网和用户终端构成一棵树状结构,这棵树的根结点位于足球比赛的现场,树叶为各个用户终端,其他中转站为该树的内部节点。
从转播站到转播站以及从转播站到所有用户终端的信号传输费用都是已知的,一场转播的总费用等于传输信号的费用总和。
现在每个用户都准备了一笔费用想观看这场精彩的足球比赛,有线电视网有权决定给哪些用户提供信号而不给哪些用户提供信号。
写一个程序找出一个方案使得有线电视网在不亏本的情况下使观看转播的用户尽可能多。
输入格式:
输入文件的第一行包含两个用空格隔开的整数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的费用。最后一行依次表示所有用户为观看比赛而准备支付的钱数。
输出格式:
输出文件仅一行,包含一个整数,表示上述问题所要求的最大用户数。
输入样例
5 3
2 2 2 5 3
2 3 2 4 3
3 4 2
输出样例
2
说明
样例解释
如图所示,共有五个结点。结点①为根结点,即现场直播站,②为一个中转站,③④⑤为用户端,共M个,编号从N-M+1到N,他们为观看比赛分别准备的钱数为3、4、2,从结点①可以传送信号到结点②,费用为2,也可以传送信号到结点⑤,费用为3(第二行数据所示),从结点②可以传输信号到结点③,费用为2。也可传输信号到结点④,费用为3(第三行数据所示),如果要让所有用户(③④⑤)都能看上比赛,则信号传输的总费用为:
2+3+2+3=10,大于用户愿意支付的总费用3+4+2=9,有线电视网就亏本了,而只让③④两个用户看比赛就不亏本了。
首先要看懂输入,,,
然后,这是道树形背包的好题
把每个节点看成一个背包
它的容量
V
V
V就是这个节点的子树大小(只包含用户)
组数为它的儿子个数
子树大小可以先预处理,也可以做背包的时候算。
再看下转移方程
f [ i ] [ j ] = m a x ( f [ i ] [ j ] , f [ i ] [ j − k ] + f [ c a ] [ k ] − e d g e [ i ] . d i s ) f[i][j]=max(f[i][j],f[i][j-k]+f[ca][k]-edge[i].dis) f[i][j]=max(f[i][j],f[i][j−k]+f[ca][k]−edge[i].dis)
其中
e
d
g
e
[
i
]
.
d
i
s
edge[i].dis
edge[i].dis是路过这条边的花费
i
i
i是最外边找与它相连的点的循环
j
j
j是枚举背包体积
k
k
k是枚举组数
f
[
i
]
[
j
]
f[i][j]
f[i][j]表示以
i
i
i为根的子树选了
j
j
j个节点的最大收益
为什么这样设呢?
当然是直觉做题多了就好了
f
[
i
]
[
j
−
k
]
f[i][j-k]
f[i][j−k]是
i
i
i这个节点少选
k
k
k个点
f
[
c
a
]
[
k
]
f[ca][k]
f[ca][k]表示
c
a
ca
ca这个节点选
k
k
k个点
然后减去这条边的花费。
既然要求用户最多
所以最后从大往小
如果有不亏本的输出就好了
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <complex>
#include <algorithm>
#include <climits>
#include <queue>
#include <map>
#include <vector>
#include <iomanip>
#define A 1000010
#define B 3010
#define ll long long
using namespace std;
struct node {
int next, to, dis;
}edge[A];
int head[A], num_edge;
void add_edge(int from, int to, int dis) {
edge[++num_edge].next = head[from];
edge[num_edge].to = to;
edge[num_edge].dis = dis;
head[from] = num_edge;
}
int f[B][B], w[B], n, m, k, a, b;
int dfs(int fr, int fa) { //dfs看不大明白可以开调试看看它怎么跑的
if (w[fr]) {
f[fr][1] = w[fr];
return 1;
}
int tot = 0;
for (int i = head[fr]; i; i = edge[i].next) {
int ca = edge[i].to;
if (ca == fa) continue;
int t = dfs(ca, fr);
tot += t;
for (int j = tot; j >= 1; j--) //子树大小
for (int k = 1; k <= t; k++) //儿子个数
f[fr][j] = max(f[fr][j], f[fr][j - k] + f[ca][k] - edge[i].dis);
}
return tot;
}
int main() {
cin >> n >> m;
for (int i = 1; i <= n - m; i++) {
cin >> k;
for (int j = 1; j <= k; j++) {
cin >> a >> b;
add_edge(i, a, b);
add_edge(a, i, b);
}
}
memset(f, -0x3f, sizeof f);
for (int i = n - m + 1; i <= n; i++) cin >> w[i];
for (int i = 1; i <= n; i++) f[i][0] = 0;
dfs(1, 0);
for (int i = m; i >= 1; i--)
if (f[1][i] >= 0) {
cout << i << endl;
break;
}
return 0;
}