LeetCode 543.二叉树的直径

543.二叉树的直径

思路：

综合前面几个二叉树的问题求解方法，最先考虑的就是递归思想。

因为考虑其解决的问题是否可以转换成子问题。

该题目需要求的是二叉树的直径，也就是相距最远的两个节点的距离。

分析得知道，任意一条最远路径都可视为由某个节点为起点，从其左节点和右节点向下遍历的同样的路径拼接而成。

即直径为：左子树的最长路径、右子树的最长路径、经过该节点的最长路径，三者中最大的一个。

经过该节点的最长路径：左子树的深度+右子树的深度

• 复杂度
  • 时间复杂度：O(n)
  • 空间复杂度：最差O(n),最优O(log(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) {
        diameter = 0;
        getMaxDepth(root);
        return diameter;
    }

    public int getMaxDepth(TreeNode node) {
        if (node == null) {
            return 0;
        }
        int left = getMaxDepth(node.left);
        int right = getMaxDepth(node.right);

        diameter = Math.max(diameter, left + right);
        return 1 + Math.max(left, right);
    }
}