hdu 1561 the more the better (树形DP)

来源:互联网 发布:phpcms建站 编辑:程序博客网 时间:2024/06/05 12:50

题目链接: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;}


0 0
原创粉丝点击