hdu 1561 the more the better （树形DP）

题目链接：hdu 1561

就树上乱搞吧~~~

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
#define maxn (210)
int w[maxn],head[maxn],fa[maxn],dp[maxn][maxn];
struct node{int v,next;}e[100000];

inline int read(){
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}

int k;
void adde(int u,int v){
	e[k].v=v; e[k].next=head[u]; head[u]=k++;
}

void find(int u,int cnt){
	dp[u][1]=w[u];
	for(int i=head[u];i!=-1;i=e[i].next){
		int v=e[i].v;
		if(cnt>1)find(v,cnt-1);
		for(int j=cnt;j>=1;j--)
			for(int k=1;k<=j;k++){
				dp[u][j]=max(dp[u][j],dp[u][k]+dp[v][j-k]);
			}
	}
}

int main(){
	int N,M;
	while(N=read()){
		M=read(); k=0;
		memset(dp,0,sizeof(dp));
		memset(head,-1,sizeof(head));
		for(int i=1;i<=N;i++){
			int a=read(); w[i]=read();
			adde(a,i); fa[i]=a;
		}
		find(0,M+1);//这里改为M+1可以免去对0的讨论
		printf("%d\n",dp[0][M+1]);
	}
	return 0;
}