题面
题目描述
某收费有线电视网计划转播一场重要的足球比赛。他们的转播网和用户终端构成一棵树状结构，这棵树的根结点位于足球比赛的现场，树叶为各个用户终端，其他中转站为该树的内部节点。
从转播站到转播站以及从转播站到所有用户终端的信号传输费用都是已知的，一场转播的总费用等于传输信号的费用总和。
现在每个用户都准备了一笔费用想观看这场精彩的足球比赛，有线电视网有权决定给哪些用户提供信号而不给哪些用户提供信号。
写一个程序找出一个方案使得有线电视网在不亏本的情况下使观看转播的用户尽可能多。
输入输出格式
输入格式：
输入文件的第一行包含两个用空格隔开的整数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的费用。最后一行依次表示所有用户为观看比赛而准备支付的钱数。
输出格式：
输出文件仅一行，包含一个整数，表示上述问题所要求的最大用户数。
输入输出样例
输入样例#1：
5 3
2 2 2 5 3
2 3 2 4 3
3 4 2
输出样例#1：
2
说明
样例解释
如图所示，共有五个结点。结点①为根结点，即现场直播站，②为一个中转站，③④⑤为用户端，共M个，编号从N-M+1到N，他们为观看比赛分别准备的钱数为3、4、2，从结点①可以传送信号到结点②，费用为2，也可以传送信号到结点⑤，费用为3（第二行数据所示），从结点②可以传输信号到结点③，费用为2。也可传输信号到结点④，费用为3（第三行数据所示），如果要让所有用户（③④⑤）都能看上比赛，则信号传输的总费用为：
2+3+2+3=10，大于用户愿意支付的总费用3+4+2=9，有线电视网就亏本了，而只让③④两个用户看比赛就不亏本了。
Solution
非常显然的树形DP. 时间复杂度看起来是(O(n^3))的, 实际上可以通过优化枚举的方法, 得到(O(n^2))的做法.
我们用f[i]表示以当前节点(非叶子节点)为根的子树中取(i)个叶子节点的最大收益, DFS处理, 考虑如何合并子节点与当前节点的信息: 我们用g[i]表示子节点中DP得到的结果, 并且将一个非叶子节点的size在开始时设定为(0).
我们用(g)数组更新(当前节点的这个子节点的f[0 ... 当前节点的size+ 这个子节点的size])的值, 然后将当前点的(size)加上子节点的(size)值.
这样, 我们合并一个子树时的时间复杂度为以其父亲节点的在它之前的所有子节点为根的子树的叶子节点数量之和, 乘上以这个子节点的叶子节点数量. 为了方便, 我们将叶子节点数量看作是整个子树的节点数量. 这样, 我们发现这个时间复杂度相当于计算以当前点为LCA的点对个数, 因此总复杂度为(O(n^2)).
#include <cstdio>
#include <cctype>
#include <vector>
#include <cstring>
#include <algorithm>
using namespace std;
namespace Zeonfai
{
inline int getInt()
{
int a = 0, sgn = 1; char c;
while (! isdigit(c = getchar())) if (c == '-') sgn *= -1;
while (isdigit(c)) a = a * 10 + c - '0', c = getchar();
return a * sgn;
}
}
const int N = 3000;
struct tree
{
struct edge
{
int v, w;
inline edge(int _v, int _w) { v = _v; w = _w; }
};
struct node
{
vector<edge> edg;
int sz;
int f[N + 7];
inline node() { edg.clear(); memset(f, -127, sizeof f); f[0] = 0; }
}nd[N + 7];
inline void addEdge(int u, int v, int w)
{
nd[u].edg.push_back(edge(v, w));
}
void DFS(int u)
{
if (nd[u].edg.empty())
{
nd[u].sz = 1;
return;
}
nd[u].sz = 0;
for (edge edg : nd[u].edg)
{
DFS(edg.v);
for (int i = nd[u].sz + nd[edg.v].sz; ~ i; -- i)
for (int j = max(0, i - nd[u].sz); j <= min(i, nd[edg.v].sz); ++ j)
nd[u].f[i] = max(nd[u].f[i], nd[u].f[i - j] + nd[edg.v].f[j] - edg.w);
nd[u].sz += nd[edg.v].sz;
}
}
inline void getAnswer()
{
for (int i = nd[1].sz; ~ i; -- i) if (nd[1].f[i] >= 0)
{
printf("%d
", i);
return;
}
}
}T;
int main()
{
#ifndef ONLINE_JUDGE
freopen("TV.in", "r", stdin);
freopen("TV.out", "w", stdout);
#endif
using namespace Zeonfai;
int n = getInt(), m = getInt();
for (int i = 1; i <= n - m; ++ i)
{
int tmp = getInt();
for (int j = 0; j < tmp; ++ j)
{
int v = getInt(), w = getInt();
T.addEdge(i, v, w);
}
}
for (int i = n - m + 1; i <= n; ++ i) T.nd[i].f[1] = getInt();
T.DFS(1);
T.getAnswer();
}