码迷,mamicode.com
首页 > 其他好文 > 详细

【POJ 2342】Anniversary party(入门树形dp)

时间:2016-08-04 06:43:50      阅读:124      评论:0      收藏:0      [点我收藏+]

标签:

dp[i][0..1]表示i不来/来参加的最大总高兴值。

则dp[i][1]+=dp[v][0](v是i的所有直接下属)

dp[i][0]+=max(dp[v][0],dp[v][1])

没有告诉你树根,dp的方向我第一次写的是用队列维护叶子节点,每次计算完,它的父亲的孩子数量--,如果减到0,就变成叶子加入队列。直到队列空。后来看别人写的两种是读入的时候维护root,还有种方法是记录父亲,然后递推祖先。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 6050
using namespace std;
int n,u,v,a[N],rt;
int dp[N][2],head[N],cnt;
struct edge{
	int to,next;
}e[N];
int DP(int u,int s){
	int &ans=dp[u][s];
	if(ans)return ans;
	if(s)ans=a[u];
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to;
		if(s)ans+=DP(v,0);
		else ans+=max(DP(v,1),DP(v,0));
	}
	return ans;
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	while(scanf("%d%d",&v,&u),u){
		e[++cnt]=(edge){v,head[u]};
		head[u]=cnt;
		if(v==rt||!rt)rt=u;
	}
	if(!rt)rt=1;//如果只有一个节点的情况
	printf("%d",max(DP(rt,1),DP(rt,0)));
}

  

【POJ 2342】Anniversary party(入门树形dp)

标签:

原文地址:http://www.cnblogs.com/flipped/p/5735202.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!