题目大意:给你一棵树,求在这棵树中最少要减去多少条边,能剩下p个节点;
自己做的时候感觉略难啊,毕竟是第一次做树形dp,没忍住看题解;
dp【i】【j】表示以i节点为根,共含有j个节点的子树需要减去的最少的边数;
这样,从树的叶子节点开始,一层一层的向上推;
对于以i为根的子树,dp[i][j]由其儿子构成,而它的每一个儿子又是由它们的儿子组成;
也就是说,假设i含有3个儿子,那么dp[i][j]可以由它的第一个儿子节点中的a个节点+第二儿子的b个节点+第三个儿子的c个节点构成(a+b+c=j)
从而我们可以把dp[i][]看成一个容量为p的背包;i的每一个儿子看做一种物品,每种物品有对应的选择数量(即为i的儿子的儿子的个数);于是就成了多重背包问题了;
在多重背包中对于一个可选择范围在0~N之间的物品,如果选0,在这就意味着,dp[i][]不选择它的这个儿子节点,此时,若果不选就要剪断它们之间的边,那么dp[i][]就要+1;如果选的数量大于一,就取最小的dp[i][j]=min(dp[i][j],dp[i][j-k]+dp[v][k])
(v是i的某一个儿子节点,dp[v][k]就是在这个儿子节点里面选k个节点,那么还需要dp[i][j-k]个)
Source Code
Problem: 1947 | User: zhuzhuxia | |
Memory: 852K | Time: 32MS | |
Language: G++ | Result: Accepted |
- Source Code
#include <iostream> #include <cstdio> #include <vector> #include <cstring> #define inf 999999999 using namespace std; int dp[222][222]; vector < int > node[222]; int visit[222]; void dfs(int root); int p; int main() { int i,j,k,l,m,n; cin>>n>>p; for(i=0;i<=n;i++) { for(j=0;j<=n;j++) { dp[i][j]=inf; } } memset(visit,0,sizeof(visit)); for(i=1;i<=n;i++) node[i].clear(); for(i=1;i<n;i++) { cin>>k>>l; node[k].push_back(l); visit[l]=1; } for(i=1;i<=n;i++) if(visit[i]==0) break; dfs(i); dp[i][p]--; int sum=inf; for(i=1;i<=n;i++) { if(dp[i][p]<sum) sum=dp[i][p]; } cout<<sum+1<<endl; return 0; } void dfs(int root) { int i,j,k,l=node[root].size(); dp[root][1]=0; for(i=0;i<l;i++) { int v=node[root][i]; dfs(v); for(j=p;j>=1;j--) { dp[root][j]+=1; for(k=1;k<j;k++) { dp[root][j]=min(dp[root][j-k]+dp[v][k],dp[root][j]); } } } }