树的直径

树的直径

题目描述

给定一棵树，求出这棵树的直径，即树上最远两点的距离。
示例1的树如下图所示。其中4到5之间的路径最长，是树的直径，距离为5+2+4=11

代码

import java.util.*;

/*
public class Interval {
  int start;
  int end;
 }
*/

public class Solution {
    /**
     * 树的直径
     * @param n int整型 树的节点个数
     * @param Tree_edge Interval类一维数组 树的边
     * @param Edge_value int整型一维数组 边的权值
     * @return int整型
     */
    int max = Integer.MIN_VALUE;
    public int solve (int n, Interval[] Tree_edge, int[] Edge_value) {
        // write code here
        //List<int[]>[] map = new ArrayList[n];
        List<int[]>[] map = new ArrayList[n];
        for(int i=0; i<n; i++){
            map[i] = new ArrayList();
        }
        //记录每个节点的分支节点和到达的权值
        for(int i=0; i<n-1; i++){
            map[ Tree_edge[i].start ].add( new int[]{Tree_edge[i].end, Edge_value[i]} );
            map[ Tree_edge[i].end ].add( new int[]{Tree_edge[i].start, Edge_value[i]} );
        }
        dfs(0, new boolean[n], map);
        return max;
    }
    /*
    主要做两件事：
        1、获取当前节点的最长路径(除了调用它的节点路径)与全局最长路径比较
        2、返回当前节点的最长路径给调用它的节点
    */
    public int dfs(int pos, boolean[] visted, List<int[]>[] map){
        //记录该节点所能达到的最大权值路径和次级权值路径
        int first_weight = 0;
        int second_weight = 0;
        visted[pos] = true;
        //循环dfs它的子节点
        for(int[] num: map[pos]){
            int children = num[0];
            int weight = num[1];
            if(visted[children]) continue;
            weight += dfs(children, visted, map);
            if(weight > first_weight){
                second_weight = first_weight;
                first_weight = weight;
            }else if(weight > second_weight){
                second_weight = weight;
            }
            max = Math.max(max, first_weight+second_weight);
        }
        return Math.max(first_weight, second_weight);
    }
}