洛谷 ~ P1273 ~ 有線電視網 (樹形揹包)
阿新 • • 發佈:2018-12-21
思路
表示該使用者願意支出的錢 表示以u為根節點的子樹有多少個使用者。 表示以為根節點,轉發到個使用者利潤的最大值 如果為葉子結點即使用者,則。
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 3005;
const int INF = 0x3f3f3f3f;
int n, m, sz[MAXN], deg[MAXN], dp[MAXN][MAXN], c[MAXN];
vector<pair<int, int> > G[MAXN];
#define X first
#define Y second
void dfs(int u, int fa)
{
dp[u][0] = 0;
if (deg[u] == 1) sz[u] = 1, dp[ u][1] = c[u];
for (auto i : G[u])
{
int v = i.X, w = i.Y;
if (v == fa) continue;
dfs(v, u);
sz[u] += sz[v];
for (int j = sz[u]; j >= 1; j--)
for (int k = 0; k <= min(sz[v], j); k++)
dp[u][j] = max(dp[u][j], dp[u][j-k] + dp[v][k] - w);
}
}
int main()
{
memset(dp, -INF, sizeof(dp));
scanf("%d%d", &n, &m);
for (int u = 1; u <= n-m; u++)
{
int k, v, w; scanf("%d", &k);
while (k--)
{
scanf("%d%d", &v, &w);
G[u].emplace_back(v, w);
G[v].emplace_back(u, w);
deg[u]++, deg[v]++;
}
}
for (int i = n-m+1; i <= n; i++) scanf("%d", &c[i]);
dfs(1, -1);
int ans = 0;
for (int i = 0; i <= sz[1]; i++)
if (dp[1][i] >= 0) ans = i;
printf("%d\n", ans);
return 0;
}
/*
5 3
2 2 2 5 3
2 3 2 4 3
3 4 2
*/