DP Intro - poj 1947 Rebuilding Roads

Posted 2020-09-21 fanbird2008

 

 

 

算法：

dp[i][j]表示以i为根的子树要变成有j个节点的状态需要减掉的边数。

考虑状态转移的时候不考虑i的父亲节点，就当不存在。最后统计最少减去边数的

时候+1。

 

 考虑一个节点时，有两种选择，要么剪掉跟子节点相连的边，则dp[i][j] = dp[i][j]+1;

 要么不剪掉，则d[i][j] = max(dp[i][j], dp[i][k]+dp[son][j-k]);

 

 

[cpp] view plain copy

 

print?

1. #include<cstdio>  
2. #include<cstring>  
3. #include<iostream>  
4. #include<vector>  
5.   
6. using namespace std;  
7.   
8. vector<int> v[160];  
9. int dp[160][160],fa[160],p,n;  
10.   
11. int minn(int a,int b)  
12. {  
13.     return a<b?a:b;  
14. }  
15.   
16. void dfs(int r,int pre)  
17. {  
18.     int s = v[r].size();  
19.     dp[r][1]=0;  
20.     for(int k=0;k<s;k++)  
21.     {  
22.         int to=v[r][k];  
23.         if(to==pre) continue;  
24.         dfs(to,r);  
25.         for(int i=p;i>=0;i--)  
26.         {  
27.             int tmp=dp[r][i]+1;  
28.             for(int j=1;j<i;j++)  
29.                 tmp=minn(tmp,dp[r][j]+dp[to][i-j]);  
30.             dp[r][i]=tmp;  
31.         }  
32.     }  
33. }  
34.   
35. int main()  
36. {  
37.     int a,b,root;  
38.     while(scanf("%d%d",&n,&p)!=EOF)  
39.     {  
40.         memset(dp,0x3f,sizeof(dp));  
41.         memset(fa,0,sizeof(fa));  
42.   
43.         for(int i=0;i<n-1;i++)  
44.         {  
45.             scanf("%d%d",&a,&b);  
46.             v[a].push_back(b);  
47.             fa[b]=a;  
48.         }  
49.         for(int i=1;i<=n;i++)  
50.         {  
51.             if(!fa[i])  
52.             {  
53.                 root=i;  
54.                 break;  
55.             }  
56.         }  
57.         dfs(root,-1);  
58.         int ans=dp[root][p];  
59.         for(int i=1;i<=n;i++)  
60.             ans=minn(dp[i][p]+1,ans);  
61.         printf("%d\n",ans);  
62.     }  
63.     return 0;  
64. }