树形dp-没有上司的舞会

原题链接
Ural大学有N名职员，编号为1~N。

他们的关系就像一棵以校长为根的树，父节点就是子节点的直接上司。

每个职员有一个快乐指数，用整数 Hi 给出，其中 1≤i≤N。

现在要召开一场周年庆宴会，不过，没有职员愿意和直接上司一起参会。

在满足这个条件的前提下，主办方希望邀请一部分职员参会，使得所有参会职员的快乐指数总和最大，求这个最大值。

输入格式
第一行一个整数N。

接下来N行，第 i 行表示 i 号职员的快乐指数Hi。

接下来N-1行，每行输入一对整数L, K,表示K是L的直接上司。

输出格式
输出最大的快乐指数。

数据范围
1≤N≤6000,
−128≤Hi≤127
输入样例：
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
输出样例：
5
思路：树形dp的入门题，其实他就是个数的最大独立集问题，题意就是找不相邻的节点让他们加起来的和最大。对于每个节点都有选和不选的状态，定义f[i] [0]表示i节点不去参加宴会，f[i] [1]表示i节点去参加宴会。那么两种状态就有了。1是根节点那么最后答案就是max(f[1] [0],f[1] [1])，对于每一个父节点都有f[u] [0]和f[u] [1]两种状态，都是从他的儿子们转移过来，对于f[u] [0]因为u不去参加舞会那么他的儿子可以去参加舞会也可以不去。则
f[u][0]=max(f[v1][0],f[v1][1])+max(f[v2][0],f[v2][1])+.......;
f[u][1]=a[u];f[u][1]=f[v1][0]+f[v2][0]+......;
转移方程出来就行了。

#include<bits/stdc++.h>
#define pb push_back//vector,deque
using namespace std;
const int N=6e3+5;
int f[N][2],a[N],n;
vector<vector<int>>e;
typedef long long ll;
int read()
{
    int x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')
            f=-1;
        ch=getchar();
    }
    while(ch>='0' && ch<='9')
        x=x*10+ch-'0',ch=getchar();
    return x*f;
}
void dfs(int u,int fa)
{
	f[u][1]=a[u];
	for(auto &v:e[u])
	{
		if(v==fa)continue;
		dfs(v,u);
		f[u][0]+=max(f[v][0],f[v][1]);
		f[u][1]+=f[v][0];
	}
}
int main()
{
     cin>>n;
     e.resize(n+1);
     for(int i=1;i<=n;i++)cin>>a[i];
     for(int i=1;i<=n-1;i++)
     {
     	int u,v;
     	cin>>u>>v;
     	e[u].push_back(v);
     	e[v].push_back(u);
	 }
	 dfs(1,-1);
	 cout<<max(f[1][0],f[1][1])<<"\n";
	return 0;
 }