poj 1947 Rebuilding Roads 树形dp

来源:互联网 发布:linux mount原理 编辑:程序博客网 时间:2024/05/30 05:40

给一颗标准的树(规定好了子父节点),让你剪掉任意条边,得到一颗独立的子树,使得这颗子树的结点数等于p,求最少剪掉几条边。

一开始的想法是先求出每个点的孩子结点数,然后枚举剪掉的边,求最小值,然后发现变成了一个排列组合的问题,复杂度是指数级。。。

然后看了网上的思路%&*$*……

/*dp[i][j]表示以i为结点生成j个结点的子树需要减掉的最少次数*//*dp[i][j]=min(dp[i.son][k]+dp[i][j-k]-2,dp[i][j]); -2是为了减去当前父节点和当前孩子结点的边*/


#include<cstdio>#include<algorithm>#include<cstring>#include<vector>using namespace std;int root;vector<int> node[200];int dp[200][200];int n,p;void cal(int fa){    for(int i=0;i<node[fa].size();i++)        cal(node[fa][i]);    for(int i=0;i<node[fa].size();i++)    for(int j=p;j>=2;j--)    {        for(int k=1;k<j;k++)        {            dp[fa][j]=min(dp[node[fa][i]][k]+dp[fa][j-k]-2,dp[fa][j]);        }    }}int main(){    int a,b;    while(scanf("%d%d",&n,&p)==2)    {        for(int i=1;i<=n;i++) node[i].clear();        root=1;        for(int i=1;i<n;i++)        {            scanf("%d%d",&a,&b);            if(b==root) root=a;            node[a].push_back(b);        }        memset(dp,0x3f,sizeof(dp));        for(int i=1;i<=n;i++)        {            dp[i][1]=node[i].size()+1;      //    printf("%d\n",dp[i][1]);        }        cal(root);        dp[root][p]--;      //根结点没有父节点,我们多算了一次。        int ans=0x3f3f3f3f;        for(int i=1;i<=n;i++)        {            ans=min(ans,dp[i][p]);        }        printf("%d\n",ans);    }    return 0;}

	
				
		
原创粉丝点击