acwing124. 二叉树中的最大路径和(树形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

题解
f[i][0]：节点0没有选，树的最大值
f[i][1]：节点0选了，树的最大值

#include<bits/stdc++.h>
using namespace std;
const int N = 6e3 + 10;
int f[N][2],a[N],in[N];
struct Edge{
    int v,next;
}edge[N];
int head[N],cnt;
void add(int u,int v){
    edge[cnt].v = v;
    edge[cnt].next = head[u];
    head[u] = cnt++;
}
int dfs(int u){
    for(int i = head[u];~i;i = edge[i].next){
        int v = edge[i].v;
        dfs(v);
        f[u][0] += max(f[v][1],f[v][0]);
        f[u][1] += f[v][0];
    }
    f[u][1] += a[u];
}
int main(){
    int n,root = -1;
    cin>>n;
    memset(head,-1,sizeof head);
    int x,y;
    for(int i = 1;i <= n;i ++)cin>>a[i];
    for(int i = 0;i < n - 1;i ++){
        cin>>x>>y;
        add(y,x);
        in[x] ++;
    }
    for(int i = 1;i <= n;i ++)
        if(in[i] == 0)root = i;
    dfs(root);
    cout<<max(f[root][0],f[root][1]);
    return 0;
}