题目要求:
思路:求平衡二叉树,就要先求出树的左右子树的高度(创建一个方法),然后判断是否满足平衡二叉树条件(另一个方法),但是这种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;
}
完整代码
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。