Virtual Judge 没有上司的舞会（树型DP）

传送门：https://vjudge.net/problem/51Nod-2605

题面：

题意：

给一颗N个节点的树，每个结点$i$都有一个价值$r[i]$
对于每个结点，如果取这个节点，那么这个节点的子节点就不能取
求能从这棵树获得的最大价值

思路：

树型DP,dfs就行
对于每个职员（结点），只有两种情况，取或不去，所以开一个二维dp；
dp[x][0]表示x结点为根的子树中x结点不去舞会能获得的最大价值
dp[x][1]表示x结点为根的子树中x结点去舞会能获得的最大价值
显然，我们需要先确定子节点的状态再来确定父节点的状态
所以这个DP应该建立在DFS上，状态转移方程：(to为u的子节点)
$dp[u][0]+=max(dp[to][1],dp[to][0])$   //u不去，它的下属可去可不去
$dp[u][1]+=dp[to][0]$           //u去，他的下属不能去

AC代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10; 
struct Edge{
	int to,next;
}e[maxn<<1];
int tot=1,n,l,k;
int r[maxn],head[maxn],dp[maxn][2];
void addEdge(int u,int v){
	e[tot]=Edge{v,head[u]};
	head[u]=tot++;
}
void dfs(int u,int f){
	dp[u][0]=0; //初始化当前结点u不去价值为0
	dp[u][1]=r[u];//初始化当节点u去的价值的r[u] 
	for(int i=head[u];i;i=e[i].next){ //dfs当前结点的子结点
		int to = e[i].to;
		if(to==f) continue;
		dfs(to,u);
		dp[u][0] += max(dp[to][0],dp[to][1]);//u不去，它的下属可去可不去 
		dp[u][1] += dp[to][0];//u去，他的下属不能去 
	}
}
int main() {
	cin>>n;
	for(int i=1; i<=n; i++) cin>>r[i];
	for(int i=1; i<n; i++) {
		cin>>l>>k;
		addEdge(l,k),addEdge(k,l); //前向星存双向边 
	}
	dfs(1,0);
	cout<<max(dp[1][0],dp[1][1])<<endl;
}