LeetCode 543.二叉树的直径

LeetCode 543.二叉树的直径_第1张图片
https://leetcode-cn.com/problems/diameter-of-binary-tree/

首先我们知道一条路径的长度为该路径经过的节点数减一,所以求直径(即求路径长度的最大值)等效于求路径经过节点数的最大值减一。

而任意一条路径均可以被看作由某个节点为起点,从其左儿子和右儿子向下遍历的路径拼接得到。
LeetCode 543.二叉树的直径_第2张图片

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
 //两结点之间的路径长度 = 一个结点到另一个结点所经过结点的个数-1
class Solution {
    int diameter = 1;
    public int diameterOfBinaryTree(TreeNode root) {
       diameter(root);
       return diameter-1;
    }
    public int diameter(TreeNode root) {
        if(root==null){
            return 0;
        }
        int l = diameter(root.left);
        int r = diameter(root.right);
        diameter = Math.max(diameter,l+r+1);
        return 1+Math.max(l,r);
    }
}

你可能感兴趣的:(LeetCode)