[洛谷-P1273]有线电视网(树形DP)
- 一、题目内容
- 题目描述
- 输入格式
- 输出格式
- 样例 #1
- 样例输入 #1
- 样例输出 #1
- 提示
- 二、思路分析
- 三、代码
一、题目内容
题目描述
某收费有线电视网计划转播一场重要的足球比赛。他们的转播网和用户终端构成一棵树状结构,这棵树的根结点位于足球比赛的现场,树叶为各个用户终端,其他中转站为该树的内部节点。
从转播站到转播站以及从转播站到所有用户终端的信号传输费用都是已知的,一场转播的总费用等于传输信号的费用总和。
现在每个用户都准备了一笔费用想观看这场精彩的足球比赛,有线电视网有权决定给哪些用户提供信号而不给哪些用户提供信号。
写一个程序找出一个方案使得有线电视网在不亏本的情况下使观看转播的用户尽可能多。
输入格式
输入文件的第一行包含两个用空格隔开的整数 N N N 和 M M M,其中 2 ≤ N ≤ 3000 2 \le N \le 3000 2≤N≤3000, 1 ≤ M ≤ N − 1 1 \le M \le N-1 1≤M≤N−1, N N N 为整个有线电视网的结点总数, M M M 为用户终端的数量。
第一个转播站即树的根结点编号为 1 1 1,其他的转播站编号为 2 2 2 到 N − M N-M N−M,用户终端编号为 N − M + 1 N-M+1 N−M+1 到 N N N。
接下来的 N − M N-M N−M 行每行表示—个转播站的数据,第 i + 1 i+1 i+1 行表示第 i i i 个转播站的数据,其格式如下:
K A 1 C 1 A 2 C 2 … A k C k K \ \ A_1 \ \ C_1 \ \ A_2 \ \ C_2 \ \ \ldots \ \ A_k \ \ C_k K A1 C1 A2 C2 … Ak Ck
K K K 表示该转播站下接 K K K 个结点(转播站或用户),每个结点对应一对整数 A A A 与 C C C , A A A 表示结点编号, C C C 表示从当前转播站传输信号到结点 A A A 的费用。最后一行依次表示所有用户为观看比赛而准备支付的钱数。单次传输成本和用户愿意交的费用均不超过 10。
输出格式
输出文件仅一行,包含一个整数,表示上述问题所要求的最大用户数。
样例 #1
样例输入 #1
5 3
2 2 2 5 3
2 3 2 4 3
3 4 2
样例输出 #1
2
提示
样例解释
如图所示,共有五个结点。结点 ① 为根结点,即现场直播站,② 为一个中转站,③④⑤ 为用户端,共 M M M 个,编号从 N − M + 1 N-M+1 N−M+1 到 N N N,他们为观看比赛分别准备的钱数为 3 3 3、 4 4 4、 2 2 2。
从结点 ① 可以传送信号到结点 ②,费用为 2 2 2;
也可以传送信号到结点 ⑤,费用为 3 3 3(第二行数据所示);
从结点 ② 可以传输信号到结点 ③,费用为 2 2 2;
也可传输信号到结点 ④,费用为 3 3 3(第三行数据所示)。
如果要让所有用户(③④⑤)都能看上比赛,则信号传输的总费用为: 2 + 3 + 2 + 3 = 10 2+3+2+3=10 2+3+2+3=10,大于用户愿意支付的总费用 3 + 4 + 2 = 9 3+4+2=9 3+4+2=9,有线电视网就亏本了,而只让 ③④ 两个用户看比赛就不亏本了。
二、思路分析
该题的题面简单的来说,就是我们在不亏本的情况下尽量多地选择叶子节点。
这道题是一道很明显的树上背包DP。这里的关键在于我们如何定义状态?
我们可以求出选择 0 0 0到 M M M个叶子节点能获得的最大利润作为数组的含义,这样做的话,我们在最后只需要去遍历一遍,如果说最大利润是大于等于0的,我们就更新答案,直到不能更新。
f [ i ] [ j ] f[i][j] f[i][j]表示在以 i i i为根节点的树中选,我们选择 j j j个叶子节点时,能够获得的最大利润。
对于树上背包的写法,作者在之前的文章中做过详细地介绍,建议去看一下:AcWing 10. 有依赖的背包问题(分组背包问题 + 树形DP)
三、代码
#include<bits/stdc++.h>
#define endl '\n'
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int N = 3e3 + 10;
vector<pii>g[N];
int val[N], f[N][N], son[N];
int n, m, k, a, c, leaves;
int get_nums(int u)
{
if(g[u].size() == 0)
{
son[u] = 0;
return 1;
}
int nums = 0;
for(int i = 0; i < g[u].size(); i ++ )
{
nums += get_nums(g[u][i].first);
}
son[u] = nums;
return nums;
}
void dp(int u)
{
if(son[u] == 0)
{
f[u][1] = val[u];
return;
}
for(int i = 0; i < g[u].size(); i ++ )
{
int s = g[u][i].first;
int spend = g[u][i].second;
dp(s);
for(int j = min(son[u], m); j >= 0; j -- )
{
for(int q = 0; q <= son[u]; q ++)
{
if(j >= q)
{
f[u][j] = max(f[u][j], f[u][j - q] + f[s][q] - spend);
}
}
}
}
}
void solve()
{
cin >> n >> m;
for(int i = 0; i < n - m; i ++ )
{
cin >> k;
for(int j = 0; j < k; j ++ )
{
cin >> a >> c;
g[i + 1].push_back({a, c});
}
}
for(int i = n - m + 1; i <= n; i ++ )
cin >> val[i];
get_nums(1);
memset(f, -INF, sizeof f);
for(int i = 1; i <= n; i ++ )
f[i][0] = 0;
dp(1);
int ans = 0;
for(int i = 0; i <= m; i ++ )
{
if(f[1][i] >= 0)
ans = i;
}
cout << ans << endl;
}
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
solve();
}