【树形DP】P1352 没有上司的舞会

题目链接：P1352 没有上司的舞会

题意：从一颗树上选点而且每个点不能直接相连，求最大值

思路：

• $dp[i][0]$ 就表示以i节点为根节点，不选i节点的最大值，$dp[i][1]$ 表示以i节点为根节点，选择i节点的最大值。
• 我们建好图，然后v为u的子节点，当u节点不选的时候，那么后面v节点选和不选我们找一个最大值 $dp[u][0]+=max(dp[v][0],dp[v][1])$ 。当u节点选的时候那么v节点肯定不能选，$dp[u][1]+=dp[v][0]$ 。

转移方程：

$$\{\begin{array}{l}dp[u][0]=\sum max(dp[s_i][0],dp[s_i][1])\\ dp[u][1]=w[u]+\sum dp[s_i][0]\end{array}$$

其中 $s_i$ 为 $u$ 的孩子，$w[i]$ 表示 $i$ 点的权重。

AC代码：

#include<iostream>
#include<vector>
using namespace std;
typedef long long LL;


vector<int>vec[6010];
int a[6010];
LL dp[6010][2];
bool st[6010];

void dfs(int u)
{
	dp[u][1]=a[u];
	
	for(int i=0;i<vec[u].size();i++)
	{
		int v=vec[u][i];
		
		dfs(v);
		
		dp[u][1]+=dp[v][0];
		dp[u][0]+=max(dp[v][0],dp[v][1]);
	}
}

int main()
{
	int n; scanf("%d",&n);
	
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	
	for(int i=1;i<n;i++)
	{
		int l,k; scanf("%d %d",&l,&k);
		
		vec[k].push_back(l);
		
		st[l]=1;
	}
	
	int root=1;
	while(st[root]) root++;
	
	dfs(root);
	
	printf("%lld",max(dp[root][0],dp[root][1]));
	
	return 0;
}