Leetcode︱543.Diameter of Binary Tree二叉树的直径.java

题目

给定一棵二叉树，你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。

示例 ：

​ 1

​ / \

​ 2 3

​ / \

4 5

返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

注意： 两结点之间的路径长度是以它们之间边的数目表示。

标签

二叉树；DFS

解题思路

方法：深度优先搜索

先计算每个节点的左右子树的深度，得出以该节点为核心的直径，经过遍历后得到最长的直径。

源自：https://leetcode-cn.com/problems/diameter-of-binary-tree/solution/er-cha-shu-de-zhi-jing-by-leetcode-solution/

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */

public class Leetcode543 {
    static int ans; //设定一个全局变量

    public static void main(String[] args) {
        TreeNode root = new TreeNode(1);
        TreeNode n1 = new TreeNode(2);
        TreeNode n2 = new TreeNode(3);
        TreeNode n3 = new TreeNode(4);
        TreeNode n4 = new TreeNode(5);
        root.left = n1;
        root.right = n2;
        root.left.left = n3;
        root.left.right = n4;
        System.out.println(diameterOfBinaryTree(root));
    }

    public static int diameterOfBinaryTree(TreeNode root) {
        depth(root);
        return ans;
    }

    //计算以node为根节点的深度(垂直节点个数)
    public static int depth(TreeNode node) {
        if (node == null) {
            return 0;
        }
        int L, R;
        L = depth(node.left); //左儿子为根的子树的深度
        R = depth(node.right); //右儿子为根的子树的深度

        //统计最长直径长度
        ans = Math.max(L + R, ans);

        //返回深度为子树最大深度+根节点的1
        return Math.max(L, R) + 1;

    }
}

来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/diameter-of-binary-tree/