POJ--1947[Rebuilding Roads] 树形DP

来源:互联网 发布:deepin linux 15.4 编辑:程序博客网 时间:2024/05/16 07:59

题意:

给你一棵树,让你求最少剪掉多少条边可以剪出一棵边数为p的子树.

 

思路:(树形DP)

(1):f[i][j]:表示以i为根(包含i),共j个节点需要切断的路的数量
(2):f[i][j+k]=min(f[i][j+k],f[i.son[t]][k])(t为i的孩子)

 

 

CODE:

/*AC代码:0ms*/#include <iostream>#include <cstdio>#include <memory.h>#include <algorithm>#include <vector>#define MAXN 155#define INF 1e8#define min(a,b) (a<b?a:b)using namespace std;int f[MAXN][MAXN];int N,P;vector<int>Tree[MAXN];void Init(){int i,j,u,v;for(i=0;i<=N;i++)Tree[i].clear();for(i=1;i<N;i++){scanf("%d%d",&u,&v);Tree[u].push_back(v);}for(i=1;i<=N;i++){for(j=1;j<=N;j++)f[i][j]=INF;}}void DFS(int root){int i,j,k;int size=Tree[root].size();for(i=0;i<size;i++)DFS(Tree[root][i]);f[root][1]=size;//类似背包:一颗颗把支添加上去for(i=0;i<size;i++){for(k=P-1;k>=1;k--){if(f[root][k]<INF){for(j=1;j<=P-1&&j+k<=P;j++){if(f[Tree[root][i]][j]<INF)f[root][k+j]=min(f[root][k+j],f[root][k]+f[Tree[root][i]][j]-1);}}}}}void Solve(){int i,res;res=INF;DFS(1);for(i=2;i<=N;i++){res=min(res,f[i][P]+1);}res=min(res,f[1][P]);printf("%d\n",res);}int main(){while(scanf("%d%d",&N,&P)!=EOF){Init();Solve();}return 0;}