树状dp经典题目

题目描述

某大学有N个职员，编号为1~N。他们之间有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。现在有个周年庆宴会，宴会每邀请来一个职员都会增加一定的快乐指数Ri，但是呢，如果某个职员的上司来参加舞会了，那么这个职员就无论如何也不肯来参加舞会了。所以，请你编程计算，邀请哪些职员可以使快乐指数最大，求最大的快乐指数。

输入输出格式

输入格式：

第一行一个整数N。(1<=N<=6000)

接下来N行，第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)

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

最后一行输入0 0

输出格式：

输出最大的快乐指数。

示例一：

输入:
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5

输出:
5

Solution:

用树形dp解决

设

f[x][0]表示以x为根的子树,且x不参加舞会的最大快乐值

f[x][1]表示以x为根的子树，且x参加了舞会的最大快乐值

则f[x][0]=dp(max(f[y][0],f[y][1])) (y是x的儿子)

f[x][1]=sigma{f[y][0]}+h[x] (y是x的儿子)

先找到唯一的树根root

则ans=max(f[root][0],f[root][1])

import java.io.*;
import java.util.*;

public class Tree {
    static int index = 0;
    static int[] value = new int[6003];
    static int[] hasFather = new int[6003];
    public static int[] h = new int[6003], ne = new int[6003], e = new int[6003];
    static int[][] f = new int[6003][2];
    static Map<Integer, Integer> map = new HashMap<>();
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        for (int i = 1; i<=n; i++) value[i] = sc.nextInt();
        //读入所有点之间的连接，记得将头节点初始化为-1，要不然dfs会爆的
        Arrays.fill(h, -1);
        for (int i = 0; i<n-1; i++) {
            int x = sc.nextInt(), y = sc.nextInt();
            insert(y, x);
            hasFather[x] = 1;
        }

        //寻找root
        int root = 1;
        for(int i = 1; i <= n; i++){
            if(hasFather[i] != 1) {
                root = i;
                break;
            }
        }
        dfs(root);
        System.out.println(Math.max(f[root][0],f[root][1]));
    }


    //插入 x 指向 y的有向边
    public static void insert(int x, int y) {
        e[index] = y;
        ne[index] = h[x];
        h[x] = index++;
    }

    public static void dfs(int x) {
        f[x][0] = 0; //x不去，那么指数初始化为0
        f[x][1] = value[x]; //x去的话，那么指数初始化为value[x]


        //遍历子树根节点的所有子节点
        for (int i = h[x]; i!=-1; i = ne[i]) {
            int y = e[i];
            dfs(y);

            // 状态转移
            f[x][0] += Math.max(f[y][0], f[y][1]);
            f[x][1] += f[y][0];
        }
    }

}