POJ 1947-Rebuilding Roads(树形背包)
题意:
一个树求得到一个节点数为p的子树,最小需要删除的边数。
分析:父节点到儿子这条边,删或不删,背包问题。
#include <map> #include <set> #include <list> #include <cmath> #include <queue> #include <stack> #include <cstdio> #include <vector> #include <string> #include <cctype> #include <complex> #include <cassert> #include <utility> #include <cstring> #include <cstdlib> #include <iostream> #include <algorithm> using namespace std; typedef pair<int,int> PII; typedef long long ll; #define lson l,m,rt<<1 #define pi acos(-1.0) #define rson m+1,r,rt<<11 #define All 1,N,1 #define read freopen("in.txt", "r", stdin) const ll INFll = 0x3f3f3f3f3f3f3f3fLL; const int INF= 0x7ffffff; const int mod = 1000000007; int dp[200][200],n,p,par[200]; int find(int x){ return x==par[x]?x:par[x]=find(par[x]); } vector<int>e[200]; void dfs(int root){ for(int i=0;i<=p;++i) dp[root][i]=INF; dp[root][1]=0; for(int i=0;i<e[root].size();++i){ int son=e[root][i]; dfs(son); for(int j=p;j>0;--j) for(int k=0;k<j;++k){ if(k==0)dp[root][j]=dp[root][j]+1;//删除边数加1 else dp[root][j]=min(dp[root][j],dp[root][j-k]+dp[son][k]);//不删背包 } } } int main() { while(~scanf("%d%d",&n,&p)){ for(int i=1;i<=n;++i){ par[i]=i; e[i].clear(); } int a,b; for(int i=0;i<n-1;++i) { scanf("%d%d",&a,&b); e[a].push_back(b); par[b]=a; } int root=find(1); dfs(root); int minn=dp[root][p]; for(int i=1;i<=n;++i){ minn=min(minn,dp[i][p]+1); } printf("%d\n",minn); } return 0; }