失眠网,内容丰富有趣,生活中的好帮手!
失眠网 > 字节跳动面试题(编程题)—平衡二叉树(思路+代码)—力扣110

字节跳动面试题(编程题)—平衡二叉树(思路+代码)—力扣110

时间:2021-02-18 22:14:49

相关推荐

字节跳动面试题(编程题)—平衡二叉树(思路+代码)—力扣110

题目要求:

思路:求平衡二叉树,就要先求出树的左右子树的高度(创建一个方法),然后判断是否满足平衡二叉树条件(另一个方法),但是这种O(n)达到了n^2 因为在求高度的时候就可能已经出现了不平衡(遍历一遍),但是得在判断时候为平衡二叉树的时候才能找到不平衡(遍历两边),遍历了两次达到了n^2,在面试中时不会通过的,所以我们可以在计算高度的时候就判断是否满足条件,这样就达到了时间复杂度是n,只遍历一遍。

1.创建一个方法用来计算左右子树的高度,为平衡树的时候才会返回高度,否则返回-1.

ps:Math.abs计算绝对值 Math.max计算最大值

public int maxDepth(TreeNode root) {if(root == null){return 0;}int leftH = maxDepth(root.left);//接收返回值 为-1时不再判断是否为平衡二叉树,会一直返回-1.int rightH = maxDepth(root.right);if(leftH >=0 && rightH >=0 && Math.abs(leftH - rightH) <= 1){return Math.max(leftH,rightH)+1;//+1返回的是根节点的高度,可以想象根节点左右子树为空时0+0+1}else{//不是平衡二叉树return -1;}

2.接收maxDepth的返回值,如果为正,说明有高度,返回true

public boolean isBalanced(TreeNode root) {if(root == null){//空树也是平衡二叉树return true;}return maxDepth(root) > 0;}

完整代码👇

class Solution {public int maxDepth(TreeNode root) {if(root == null){return 0;}int leftH = maxDepth(root.left);int rightH = maxDepth(root.right);if(leftH >=0 && rightH >=0 && Math.abs(leftH - rightH) <= 1){return Math.max(leftH,rightH)+1;}else{return -1;}}public boolean isBalanced(TreeNode root) {if(root == null){//空树也是平衡二叉树return true;}return maxDepth(root) > 0;}}

运行结果:图①O(n*n) ,图②O(n)

如果觉得《字节跳动面试题(编程题)—平衡二叉树(思路+代码)—力扣110》对你有帮助,请点赞、收藏,并留下你的观点哦!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。