LeetCode算法题-Diameter of Binary Tree(Java实现)
这是悦乐书的第257次更新,第270篇原创
01 看题和准备
今天介绍的是LeetCode算法题中Easy级别的第124题(顺位题号是543)。给定二叉树,您需要计算树的直径长度。 二叉树的直径是树中任意两个节点之间最长路径的长度。 此路径可能会也可能不会通过根节点。例:
给出一棵二叉树
1
/ \
2 3
/ \
4 5
返回3,这是路径[4,2,1,3]或[5,2,1,3]的长度。
注意:两个节点之间的路径长度由它们之间的边数表示。
本次解题使用的开发工具是eclipse,jdk使用的版本是1.8,环境是win7 64位系统,使用Java语言编写和测试。
02 解题
题目要求是求二叉树的直径,根据题目给的示例,也可以理解成从根节点开始,左子树的深度与右子树的深度之和,但是题目又说了,也可能不经过根节点,所以我们需要求两者的最大值。计算任意一个节点的最长路径值,在其中取出最大值即可。递归方法的主体,和求二叉树最大深度的代码一样,只是中间加了一步取最大值的操作而已。
此解法的时间复杂度是O(n),空间复杂度是O(n)。
public int diameterOfBinaryTree(TreeNode root) {
helper(root);
return max;
}
private int max = 0;
public int helper(TreeNode root) {
if (root == null) {
return 0;
}
int left = helper(root.left);
int right = helper(root.right);
max = Math.max(max, left+right);
return Math.max(left, right)+1;
}
03 小结
算法专题目前已日更超过三个月,算法题文章124+篇,公众号对话框回复【数据结构与算法】、【算法】、【数据结构】中的任一关键词,获取系列文章合集。
以上就是全部内容,如果大家有什么好的解法思路、建议或者其他问题,可以下方留言交流,点赞、留言、转发就是对我最大的回报和支持!