[java]树形DP

树形DP

树形 DP，即在树上进行的 DP。由于树固有的递归性质，树形 DP 一般都是递归进行的。OIWIKI

1245 树的直径

给你这棵「无向树」，请你测算并返回它的「直径」：这棵树上最长简单路径的 边数。

我们用一个由所有「边」组成的数组 edges 来表示一棵无向树，其中 edges[i] = [u, v] 表示节点 u 和 v 之间的双向边。

树上的节点都已经用 {0, 1, …, edges.length} 中的数做了标记，每个节点上的标记都是独一无二的。

class Solution {
    private List<Integer>[] g;
    private int ans = 0;
    public int treeDiameter(int[][] edges) {
        int n = edges.length;
        g = new ArrayList[n + 1];
        Arrays.setAll(g, e -> new ArrayList<>());
        for(int[] e : edges){
            int x = e[0], y = e[1];
            g[x].add(y);
            g[y].add(x);
        }
        dfs(0, -1);//fa:无向边防止递归到父节点
        return ans;
    }

    private int dfs(int x, int fa){
        int d1 = 0, d2 = 0;
        for(int y : g[x]){
            if(y == fa) continue;
            int d = dfs(y, x) + 1;
            if(d > d1){
                d2 = d1;
                d1 = d;
            }else if(d > d2){
                d2 = d;
            }
            ans = Math.max(ans, d1 + d2);
        }
        return d1;
    }
}