POJ 2342 （树状DP）入门

入门比较好的博客

#include "cstdio"
#include "iostream"
#include "cstring"
#include "algorithm"
using namespace std;
const int MAX=6005;
/*
树形dp 入门 
题意：公司举办 晚会，为了提高活跃度（每个人都有自己的活跃度），
下属不能和直接上司一起出现，问最大活跃度
题解： dp[i][0] 第i个人不来 初始化为  0 
	   dp[i][1] 第i个人来   初始化为活跃度
从叶子节点往上积累，
// 上司来，下属不来
dp[root][1] += dp[i][0];
// 上司不来，下属来，也可能不来 ，选最大 
dp[root][0] += max(dp[i][0],dp[i][1]);

*/
int n;
int dp[MAX][2],father[MAX];
// dp[i][0] 表示不去
// dp[i][1] 表示去 
int vis[MAX]; 
int tree_dp(int root)
{
	int i;
	vis[root]=1; 
	for(int i=1;i<=n;i++)
	{// 遍历所有点 
		if(!vis[i] && father[i]== root)
		{// i是root的下属 
			tree_dp(i); // 找到了root下属 ，找到叶子，从叶子节点往上dp
			dp[root][1] += dp[i][0];// 上司来，下属不来
			dp[root][0] += max(dp[i][0],dp[i][1]);// 上司不来，下属来，也可能不来 ，选最大 
		}
	}
}
int main()
{
	while(~scanf("%d",&n))
	{
		memset(dp,0,sizeof(dp));
		memset(father,0,sizeof(father));
		memset(vis,0,sizeof(vis));
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&dp[i][1]);//每个去的人的价值 
		}
		int root=0;// 记录父节点
		int u,v;
		while(scanf("%d%d",&u,&v) &&u+v)
		{
			father[u]=v;
			//若根节点是u，那么将根节点转到v 
			if(root == u)
				root=v;
		}
		//查找父节点, 
		while(father[root])
			root=father[root];
		tree_dp(root);
		int ans=max(dp[root][1],dp[root][0]); 
		printf("%d\n",ans);
	}
	return 0;
}