树的直径

/*
 * Leetcode 1245: Tree Diameter
 * 
 * 思路：
 * 1. 将树转换为图，并使用DFS进行两次遍历。
 * 2. 第一次DFS找到从任意节点开始的最远节点A。
 * 3. 第二次DFS从节点A开始，找到最远的节点B，
 * 这样A到B的路径就是树的直径。
 */

import java.util.*;
import java.util.stream.*;

class Solution {
    private int maxDiameter = 0;
    
    public int treeDiameter(int[][] edges) {
        // 构建图
        Map<Integer, List<Integer>> graph = Arrays.stream(edges)
                .flatMap(edge -> Stream.of(new int[][]{{edge[0], edge[1]}, {edge[1], edge[0]}}))
                .collect(Collectors.groupingBy(
                    e -> e[0], 
                    Collectors.mapping(e -> e[1], Collectors.toList())
                ));
        
        // 第一次DFS
        int[] firstDFS = dfs(0, -1, graph);
        
        // 第二次DFS
        int[] secondDFS = dfs(firstDFS[1], -1, graph);
        
        return secondDFS[0];
    }
    
    private int[] dfs(int node, int parent, Map<Integer, List<Integer>> graph) {
        return graph.getOrDefault(node, Collections.emptyList()).stream()
                .filter(neighbor -> neighbor != parent)
                .map(neighbor -> dfs(neighbor, node, graph))
                .max(Comparator.comparingInt(result -> result[0]))
                .map(result -> new int[]{result[0] + 1, result[1]})
                .orElse(new int[]{0, node});
    }
}