543. 二叉树的直径(简单)

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

示例 :

给定二叉树

      1
     / \
    2   3
   / \     
  4   5    

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/diameter-of-binary-tree

思路:

最大值不一定包含根节点,但是一定是:经过一个节点,该节点左右子树的最大深度之和 +1(二叉树的根节点深度为 0)
于是,可以使用 DFS,找出所有节点的最大直径,在取出最大值 res;

定义一个全局变量 res,用来记录最大直径
使用 dfs(root) 遍历所有的节点,dfs(root) 的作用是:找出以 root 为根节点的二叉树的最大深度

将根节点的深度定义为 1
root 为跟节点的最大深度为 Math.max(leftDepth,rigthDepth) + 1
res 取值为以经过 root,左右子树的最大深度之和 leftDepth + rigthDepth(不用加 1,是因为根节点的深度是 1)
通过递归,找到 res 的最大值

作者:sugar-31
链接:https://leetcode-cn.com/problems/diameter-of-binary-tree/solution/java-shen-du-you-xian-bian-li-dfs-by-sugar-31/

类比leetcode687

代码1:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
int res =0;
public int diameterOfBinaryTree(TreeNode root) {
dfs(root);
return res;
}
public int dfs(TreeNode root){
if(root==null)
return 0;
int left = dfs(root.left);
int right = dfs(root.right);
res = Math.max(res,left+right);
return Math.max(left,right)+1;
}
}