【剑指offer-39】平衡二叉树
- 考点:树
- 时间限制:1秒
- 空间限制:32768K
- 输入一棵二叉树,判断该二叉树是否是平衡二叉树。
思路:
平衡二叉树(Balanced Binary Tree)具有以下性质:它是一棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。
思路1和思路2都放在代码里面了。
思路1:最直接的做法,遍历每个结点,借助一个获取树深度的递归函数,根据该结点的左右子树高度差判断是否平衡,然后递归地对左右子树进行判断。
这种做法有很明显的问题,在判断上层结点的时候,会多次重复遍历下层结点,增加了不必要的开销。如果改为从下往上遍历,如果子树是平衡二叉树,则返回子树的高度;如果发现子树不是平衡二叉树,则直接停止遍历,这样至多只对每个结点访问一次。
这就是思路2。
后续遍历时,遍历到一个节点,其左右子树已经遍历 依次自底向上判断,每个节点只需要遍历一次
代码:
思路一:
public class Solution {
public boolean IsBalanced_Solution(TreeNode root) {
if (root == null) {
return true;
}
int left = getDepth(root.left);
int right = getDepth(root.right);
return Math.abs(left - right) <= 1 && IsBalanced_Solution(root.right) && IsBalanced_Solution(root.left);
}
public int getDepth(TreeNode root) {
if (root == null) {
return 0;
}
int left = getDepth(root.left);
int right = getDepth(root.right);
return Math.max(left, right) + 1;
}
}
思路2:
public class Solution {
private boolean isBalanced = true;
public boolean IsBalanced_Solution(TreeNode root) {
getDepth(root);
return isBalanced;
}
public int getDepth(TreeNode root) {
if (root == null) {
return 0;
}
int left = getDepth(root.left);
int right = getDepth(root.right);
if (Math.abs(left - right) > 1) {
isBalanced = false;
}
return Math.max(left, right) + 1;
}
}
我的问题:
在判断条件中添加条件,可以减少递归次数。
public int getDepth(TreeNode root) {
if (root == null || isBalanced == false) { // 添加isBalanced的判断条件,减少递归次数
return 0;
}
// ...