友情支持
如果您觉得这个笔记对您有所帮助,看在D瓜哥码这么多字的辛苦上,请友情支持一下,D瓜哥感激不尽,😜
有些打赏的朋友希望可以加个好友,欢迎关注D 瓜哥的微信公众号,这样就可以通过公众号的回复直接给我发信息。
公众号的微信号是: jikerizhi 。因为众所周知的原因,有时图片加载不出来。 如果图片加载不出来可以直接通过搜索微信号来查找我的公众号。 |
543. Diameter of Binary Tree
Given a binary tree, you need to compute the length of the diameter of the tree. The diameter of a binary tree is the length of the longest path between any two nodes in a tree. This path may or may not pass through the root.
Example:
Given a binary tree
1
/ \
2 3
/ \
4 5
Return 3, which is the length of the path [4,2,1,3] or [5,2,1,3].
Note: The length of path between two nodes is represented by the number of edges between them.
解题分析
这道题就是一个深度优先搜索。不同之处在于,要在遍历过程中,把直径的最大值保存下来。同时,还要返回子树中最长的路径。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
/**
* Runtime: 0 ms, faster than 100.00% of Java online submissions for Diameter of Binary Tree.
* Memory Usage: 39 MB, less than 19.48% of Java online submissions for Diameter of Binary Tree.
*/
private int result;
public int diameterOfBinaryTree(TreeNode root) {
result = 1;
dfs(root);
return result - 1;
}
private int dfs(TreeNode root) {
if (Objects.isNull(root)) {
return 0;
}
int left = dfs(root.left);
int right = dfs(root.right);
result = Math.max(result, left + right + 1);
return Math.max(left, right) + 1;
}