思路
我们转化一下,假设把所有的边先都删除,那么我们要考虑的就是这条边是不是要加进去,这就是一个01背包+树形dp。
定义deg表示
u
u
u点的度数。
定义
d
p
[
u
]
[
j
]
dp[u][j]
dp[u][j]表示以
u
u
u为根保留
j
j
j个结点所需要删除的边数。
d
p
[
u
]
[
1
]
=
d
e
g
[
u
]
dp[u][1]= deg[u]
dp[u][1]=deg[u]
d
p
[
u
]
[
j
]
=
m
i
n
(
d
p
[
u
]
[
j
]
,
d
p
[
u
]
[
i
]
+
d
p
[
v
]
[
j
−
i
]
−
2
)
,
(
j
:
p
−
>
2
)
(
i
:
1
−
>
j
−
1
)
,
v
是
u
的
儿
子
dp[u][j] = min(dp[u][j],dp[u][i]+dp[v][j-i]-2),(j: p->2) \quad (i:1->j-1),v是u的儿子
dp[u][j]=min(dp[u][j],dp[u][i]+dp[v][j−i]−2),(j:p−>2)(i:1−>j−1),v是u的儿子
为什么要 -2 呢?因为
d
p
[
u
]
[
i
]
dp[u][i]
dp[u][i]中把u->v这条边删掉了,
d
p
[
v
]
[
j
−
i
]
dp[v][j-i]
dp[v][j−i]中也把u->v这条边删掉了。
a
n
s
=
m
i
n
(
d
p
[
1
−
>
n
]
[
p
]
)
ans=min(dp[1->n][p])
ans=min(dp[1−>n][p])
记得初始化dp为INF
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 155;
const int INF = 0x3f3f3f3f;
int n, p, deg[MAXN], dp[MAXN][MAXN];
vector<int> G[MAXN];
void dfs(int u, int fa)
{
dp[u][1] = deg[u];
for (auto v : G[u])
{
if (v == fa) continue;
dfs(v, u);
for (int j = p; j >= 2; j--)
for (int i = 1; i < j; i++)
dp[u][j] = min(dp[u][j], dp[u][i] + dp[v][j-i] - 2);
}
}
int main()
{
scanf("%d%d", &n, &p);
memset(dp, INF, sizeof(dp));
for (int i = 0; i < n-1; i++)
{
int u, v; scanf("%d%d", &u, &v);
G[u].push_back(v);
G[v].push_back(u);
deg[u]++, deg[v]++;
}
dfs(1, -1);
int ans = INF;
for (int i = 1; i <= n; i++) ans = min(ans, dp[i][p]);
printf("%d\n", ans);
return 0;
}
/*
11 6
1 2
1 3
1 4
1 5
2 6
2 7
2 8
4 9
4 10
4 11
*/