543. 二叉树的直径(java)
·
题目描述
给你一棵二叉树的根节点,返回该树的 直径 。
二叉树的 直径 是指树中任意两个节点之间最长路径的 长度 。这条路径可能经过也可能不经过根节点 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; // 返回该节点为根的子树的深度
}
}
更多推荐


所有评论(0)