标签：
http://blog.csdn.net/dellaserss/article/details/8799730
这题其实和上一题思路是一样的，一个0节点作为根节点，通过剩余量来遍历子树。
#include <iostream> #include <string> #include <cstring> #include <cstdlib> #include <cstdio> #include <cmath> #include <algorithm> #include <stack> #include <queue> #include <cctype> #include <vector> #include <iterator> #include <set> #include <map> #include <sstream> using namespace std; #define mem(a,b) memset(a,b,sizeof(a)) #define pf printf #define sf scanf #define spf sprintf #define pb push_back #define debug printf("!\n") #define MAXN 205 #define MAX(a,b) a>b?a:b #define blank pf("\n") #define LL long long #define ALL(x) x.begin(),x.end() #define INS(x) inserter(x,x.begin()) #define pqueue priority_queue #define INF 0x3f3f3f3f int n,m; struct node { int y,next; }tree[1000]; int head[MAXN],dp[MAXN][MAXN],ptr=1; int sum[MAXN],vl[MAXN],vis[MAXN]; void add(int x,int y) { tree[ptr].y = y; tree[ptr].next = head[x]; head[x] = ptr++; } void dfs(int root) { if(vis[root]) return; int i,j,k; vis[root] = sum[root] = 1; int tot = 0; for(i=head[root]; i!=-1; i=tree[i].next) { int p = tree[i].y; if(!vis[p]) { dfs(p); sum[root]+=sum[p]; tot++; //pf("i%d p%d tot%d sum%d\n",root,p,tot,sum[root]); } } dp[root][1] = vl[root]; for(i=head[root]; i!=-1; i=tree[i].next) { int p = tree[i].y; for(j = sum[root];j>1;j--) { for(k = 1;k<j;k++) { if(dp[root][k]!=-1 && dp[p][j-k]!=-1) { dp[root][j] = max(dp[root][j],dp[root][k]+dp[p][j-k]); //pf("i%d j%d k%d p%d dp%d\n",root,j,k,p,dp[root][j]); } } } } } int main() { int i,j,k,a,b; while(~sf("%d%d",&n,&m) && m+n>0) { mem(head,-1); ptr = 1; for(i=1;i<=n;i++) { sf("%d%d",&a,&b); add(i,a); add(a,i); vl[i] = b; } mem(dp,-1); mem(vis,0); dfs(0); pf("%d\n",dp[0][m+1]); } return 0; }
但我发现这道题因为要统计子节点数量，其实用邻接表更方便一些：（但速度会慢很多）
#include <iostream> #include <string> #include <cstring> #include <cstdlib> #include <cstdio> #include <cmath> #include <algorithm> #include <stack> #include <queue> #include <cctype> #include <vector> #include <iterator> #include <set> #include <map> #include <sstream> using namespace std; #define mem(a,b) memset(a,b,sizeof(a)) #define pf printf #define sf scanf #define spf sprintf #define pb push_back #define debug printf("!\n") #define MAXN 205 #define MAX(a,b) a>b?a:b #define blank pf("\n") #define LL long long #define ALL(x) x.begin(),x.end() #define INS(x) inserter(x,x.begin()) #define pqueue priority_queue #define INF 0x3f3f3f3f int n,m; struct node { int y,next; }tree[1000]; int head[MAXN],dp[MAXN][MAXN],ptr=1; int sum[MAXN],vl[MAXN],vis[MAXN]; vector<int> son[1000]; void dfs(int root) { int i,j,k; for(i=0;i<son[root].size();i++) { int p = son[root][i]; if(son[p].size()>0) dfs(p); for(j=m;j>1;j--) { for(k=1;k<j;k++) { dp[root][j] = max(dp[root][j],dp[root][k] + dp[p][j-k]); //pf("i%d j%d k%d p%d dp%d\n",root,j,k,p,dp[root][j]); } } } } int main() { int i,j,k,a,b; while(~sf("%d%d",&n,&m) && m+n>0) { mem(dp,0); m++; for(i=0;i<=n;i++) son[i].clear(); for(i=1;i<=n;i++) { sf("%d%d",&a,&b); son[a].pb(i); for(j=1;j<=m;j++) dp[i][j] = b; } dfs(0); pf("%d\n",dp[0][m]); } return 0; }
标签：
原文地址：http://www.cnblogs.com/qlky/p/5655556.html
