二叉树任意两个节点间的最大距离(Java,LeetCode 543二叉树的直径 递归)

文章目录

题目

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

示例

给定二叉树:

     1
    / \
   2   3
  / \     
 4   5    

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

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

代码

思路:
使用递归的方法,用一个全局变量max保存最大距离
(一开始没写出来是因为把自己绕进去了,重点就在于,递归的时候,更新maxleftMax+rightMax,而返回leftMax和rightMax的较大值

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    int max = 0;
    public int diameterOfBinaryTree(TreeNode root) {
        getMax(root);
        return max;

    }
    public int getMax(TreeNode node){
        if(node == null){
            return 0;
        }
        int leftMax = 0;
        int rightMax = 0;
        if(node.left != null){
            leftMax = getMax(node.left);
        }
        if(node.right != null){
            rightMax = getMax(node.right);
        }
        max = Math.max(leftMax + rightMax, max);
        return Math.max(leftMax, rightMax) + 1;
    }
}
©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页