http://poj.org/problem?id=1947
题意:输入 n, m 表示一棵树有 n 个顶点,然后输入 n-1 条边,要求删除一些边使得树中剩下 m 个点,问最少要删除多少边。
思路:树形dp,转移方程是:
1.不加子树son,dp[rt][i] = dp[rt][i] + 1 (不加子树son,相当于删去一条边)。
2.加子树son,dp[rt][i] = min(dp[rt][j] + dp[son][i - j]) (以s为根的树的结点数加上其子树的结点数等于i)
最后答案就是在dp[i][m]中取小,要注意的一点是,如果i不是根,值还需要+1,因为要脱离原来的根,还要去掉一条边。
#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
using namespace std;
int n, m;
int dp[155][155];
vector<int>tr[155];
int Dfs(int rt, int pre)
{
int sum = 1;
dp[rt][1] = 0;
int len = tr[rt].size();
for(int i = 0; i < len; i++){
int son = tr[rt][i];
if(son == pre) continue;
sum += Dfs(son, rt);
for(int k = sum; k >= 1; k--)
{
dp[rt][k] = dp[rt][k] + 1; //截断子树son
for(int l = 1; l < k; l++)
{
if(dp[rt][k] > dp[rt][k-l] + dp[son][l]){
dp[rt][k] = dp[rt][k-l] + dp[son][l]; //加入son
}
}
}
}
return sum;
}
int main()
{
while(cin>>n>>m){
int x, y;
memset(dp, 127, sizeof(dp)); //初始化为0x7f7f7f7f
for(int i = 0; i <= n; i++){
dp[i][0] = 1;
tr[i].clear();
}
for(int i = 1; i < n; i++){
cin>>x>>y;
tr[x].push_back(y);
tr[y].push_back(x);
}
Dfs(1, -1);
int minn = dp[1][m];
for(int i = 2; i <= n; i++) //不要父结点
{
if(dp[i][m] + 1 < minn){
minn = dp[i][m] + 1;
}
}
cout<<minn<<endl;
}
return 0;
}