HDU 1520 Anniversary party 简单树状DP

http://acm.hdu.edu.cn/showproblem.php?pid=1520

题意：给出一颗树，以及每个点的值，选出一些点，满足任一点的直接前驱和直接后继不与此点同时出现，求这些点的和的最大值；

思路：对任一点，以此点为根节点的树的最大值为max（选择此点+所有儿子节点都不选的最大值，不选此点+所有儿子节点选或不选的最大值）；

#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#include <algorithm>
#include <map>
#include <vector>
#include <stack>
using namespace std;
#define M 6005
#define ll long long
#define int64 __int64

struct node
{
	int yes , no;
}arr[M];
int n , con[M] , is[M];
vector<int> son[M];

void Dfs(int root)
{
	//初始化当前点的值
	arr[root].no = 0;
	arr[root].yes = con[root];

	if (son[root].size() <= 0)return;//若是叶子返回

	int i , up = son[root].size();
	for (i = 0 ; i < up ; i++)//先给儿子节点赋值再计算当前点的值
	{
		Dfs(son[root][i]);
		arr[root].yes += arr[son[root][i]].no;
		arr[root].no += max(arr[son[root][i]].no,arr[son[root][i]].yes);
	}
}

int main()
{
	while (~scanf("%d",&n))
	{
		int i , ans = 0 , l , k;
		for (i = 1 ; i <= n ; i++)scanf("%d",con+i);
		memset(is , 0 , sizeof is);
		while (scanf("%d%d",&l,&k) , l)//输入边
		{
			son[k].push_back(l);
			is[l] = 1;
		}
		for (i = 1 ; i <= n ; i++)//寻找根节点
		{
			if (!is[i])
			{
				Dfs(i);
				ans += max(arr[i].yes,arr[i].no);
			}
		}
		printf("%d\n",ans);
		for (i = 1 ; i <= n ; i++)son[i].clear();
	}
	return 0;
}