牛客网刷题-树的直径

问题描述

给定一棵树，求出这棵树的直径，即树上最远两点的距离。

输入描述:
输入节点个数，每条边的两个节点，边的权重

输出描述:
输出最长的路径

示例

示例1

输入
6,[[0,1],[1,5],[1,2],[2,3],[2,4]],[3,4,2,1,5]

输出
11

解决思路

分析

本题主要是通过深度优先遍历，通过对比判断走每一条路径的权重值，得到最大值

方法

1. 采用深度优先遍历的方式，遍历树，得到最大值

代码实现

public class Solution {
    /**
     * 树的直径
     *
     * @param n int整型 树的节点个数
     * @param Tree_edge Interval类一维数组 树的边
     * @param Edge_value int整型一维数组 边的权值
     * @return int整型
     */
    //6,[[0,1],[1,5],[1,2],[2,3],[2,4]],[3,4,2,1,5]
    int res = 0;

    public int solve(int n, Interval[] Tree_edge, int[] Edge_value) {
        // write code here
        // list数组存储每个节点的所有边
        List<int[]>[] map = new ArrayList[n];
        // 将所有的节点初始化
        for (int i = 0; i < n; i++) map[i] = new ArrayList();
        // 遍历所有的边，一次添加到对应数组的list中
        for (int i = 0; i < Tree_edge.length; 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]});
        }
        // 深度优先遍历，通过数组的bool值判断节点是否已经访问过
        dfs(map, 0, new boolean[n]);
        return res;
    }

    private int dfs(List<int[]>[] map, int index, boolean[] visited) {
        // 将节点置为已经访问
        visited[index] = true;
        // 获取节点的所有的边
        List<int[]> list = map[index];
        int weight1 = 0, weight2 = 0;
        // 遍历所有的边
        for (int[] child2weight : list) {
            // 获取边的另一端
            int child = child2weight[0];
            // 获取边的权重
            int weight = child2weight[1];
            // 判断边是否已经访问
            if (visited[child]) continue;
            // 计算路径长度
            int num = weight + dfs(map, child, visited);

            if (num > weight1) {
                weight2 = weight1;
                weight1 = num;
            } else if (num > weight2) {
                weight2 = num;
            }
            if (weight1 + weight2 > res) res = weight1 + weight2;
        }
        return Math.max(weight1, weight2);
    }
}

小伙伴如果想测试的话，可以直接到牛客网这个链接做测试

树的直径-牛客网