原题链接
显然是一个树形\(DP\),状态转移则是裸的分组背包模型。
先将边权转换为点权,即每个转播台或用户的权值为原来的点权减去它到父亲的边权(没有的作为\(0\)去计算),记为\(V[x]\)。
设\(f[x][k]\)表示以\(x\)为根的子树中,选择\(k\)个用户所能达到的最大收入,\(size[x]\)表示以\(x\)为根的子树中用户的个数。
则对于\(x\)的一个儿子\(y\),有状态转移方程:
\(\qquad\qquad j = size[x] \to 0, f[x][j] = \max\limits ^ {k = j \to 0}\{ f[x][j], f[x][j - k] + f[y][k] \}\)
在转移完毕时,再将\(x\)自身的权值加上:\(i = size[x] \to 1, f[x][i] = f[x][i] + V[x]\)
初始化则是先将\(f\)全部赋为极小值,然后在\(DFS\)中赋初值:
- 不论\(x\)是什么,\(f[x][0] = 0\)。
- 若\(x\)为用户,则\(f[x][1] = V[x]\)。
最后的答案就是满足\(f[x][k] \geqslant 0\)中最大的\(k\)。
虽然这个方法理论上是\(O(n^3)\)的,但因为大多数都是冗余的状态,且实际实现的常数一般很小,所以是能够通过的。
#include<cstdio>
#include<cstring>
using namespace std;
const int N = 3010;
const int M = N << 1;
int fi[N], di[M], ne[M], f[N][N], V[N], si[N], l, p;
inline int re()
{
int x = 0;
char c = getchar();
bool p = 0;
for (; c < '0' || c > '9'; c = getchar())
p |= c == '-';
for (; c >= '0' && c <= '9'; c = getchar())
x = x * 10 + c - '0';
return p ? -x : x;
}
inline void add(int x, int y)
{
di[++l] = y;
ne[l] = fi[x];
fi[x] = l;
}
inline int maxn(int x, int y)
{
return x > y ? x : y;
}
void dfs(int x, int fa)
{
int i, j, k, y;
f[x][0] = 0;
if (x > p)
{
si[x] = 1;
f[x][1] = V[x];
return;
}
for (i = fi[x]; i; i = ne[i])
if ((y = di[i]) ^ fa)
{
dfs(y, x);
si[x] += si[y];
for (j = si[x]; ~j; j--)
for (k = j; ~k; k--)
f[x][j] = maxn(f[x][j], f[x][j - k] + f[y][k]);
}
for (i = si[x]; i; i--)
f[x][i] += V[x];
}
int main()
{
int i, j, k, n, m, x, y;
n = re();
m = re();
p = y = n - m;
for (i = 1; i <= y; i++)
for (k = re(), j = 1; j <= k; j++)
{
x = re();
add(i, x);
add(x, i);
V[x] = -re();
}
for (i = 1; i <= m; i++)
V[y + i] += re();
memset(f, 250, sizeof(f));
dfs(1, 0);
for (i = n; i; i--)
if (f[1][i] >= 0)
break;
printf("%d", i);
return 0;
}