题目描述

给你一棵二叉树的根节点,返回该树的 直径 。

二叉树的 直径 是指树中任意两个节点之间最长路径的 长度 。这条路径可能经过也可能不经过根节点 root 。

两节点之间路径的 长度 由它们之间边数表示。

示例 1:

输入:root = [1,2,3,4,5]
输出:3
解释:3 ,取路径 [4,2,1,3] 或 [5,2,1,3] 的长度。

示例 2:

输入:root = [1,2]
输出:1

代码思路:

  • 直径计算:对于每个节点,直径候选值 = 左深度 + 右深度

  • 深度计算:当前节点深度 = max(左深度, 右深度) + 1

  • 全局更新:在递归过程中动态维护最大值

/**
 * 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;
 *     }
 * }
 */
class Solution {
    public int diameterOfBinaryTree(TreeNode root) {
        int[] maxd = new int[1];
        max_Deoth(root, maxd);
        return maxd[0];
    }
    public int max_Deoth(TreeNode root,int[] max_diame){
        if(root==null){
            return 0;
        }
        if(root.left==null&&root.right==null){
            return 1;
        }
        int l_len = max_Deoth(root.left,max_diame);
        int r_len = max_Deoth(root.right,max_diame);
        int d =0;
        if(l_len==0 && r_len!=0){
            d=r_len;
        }
        if(r_len==0 && l_len!=0){
            d=l_len;
        }
        if(r_len!=0 && l_len!=0){
            d = l_len+r_len;
        }   
        if(max_diame[0]<d){
            max_diame[0] = d;
        }
        return l_len>r_len?l_len+1:r_len+1;
    }
}

官方:

class Solution {
    int ans;
    public int diameterOfBinaryTree(TreeNode root) {
        ans = 1;
        depth(root);
        return ans - 1;
    }
    public int depth(TreeNode node) {
        if (node == null) {
            return 0; // 访问到空节点了,返回0
        }
        int L = depth(node.left); // 左儿子为根的子树的深度
        int R = depth(node.right); // 右儿子为根的子树的深度
        ans = Math.max(ans, L+R+1); // 计算d_node即L+R+1 并更新ans
        return Math.max(L, R) + 1; // 返回该节点为根的子树的深度
    }
}

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐