543. 二叉树的直径

Posted 2021-12-04 tu9oh0st

题目描述

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

示例 :
给定二叉树

          1
         /         2   3
       / \     
      4   5    

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

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

分析

感觉这个题最精髓的就是这两行代码了， 因为题目要求我们算的直径考虑的是最长的边，而不是最长的 path 的node数目，如果是统计最长的 path 的 node 数目，每次递归的时候加 1 就好了，但是因为这里算的是边长， 所以在走到叶子节点的时候统计的长度是 0， 如果不是叶子节点就是1加上下一层调用该函数的返回值。

int left = dfs(root.left);
int right = dfs(root.right);

时间复杂度分析：o(n)

贴出代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
   int diameter;
    public int diameterOfBinaryTree(TreeNode root) {
        if (root == null){
            return 0;
        }
        dfs(root);
        return diameter - 1;
    }

    
    private int dfs(TreeNode root){
        if (root == null){
            return 0;
        }
        int left = dfs(root.left);
        int right = dfs(root.right);
        diameter = Math.max(diameter,left + right + 1);
        return Math.max(left,right) + 1;
    }
}