题目
输入一棵二叉树的根结点,判断该树是不是平衡二叉树。如果某二叉树中任意结点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
思路
首先,明确平衡二叉树的定义:它是一棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。
- 思路一,在55-1:树的深度的基础上,遍历每个节点,并判断每个节点的左右子树高度差是否超过1,若超过,则不是平衡二叉树;若所有节点都不超过,则是平衡二叉树。
- 思路二,上述思路显然一个节点会被遍历多次。因此思路二给出只遍历一次的思路(且有剪枝)。即使用后序遍历的方式遍历二叉树的节点,这样遍历父节点时已经遍历过其左右子树,只需要记录下每个节点的深度,即可在一次后序遍历后,知道这棵树是否平衡。
测试用例
1.功能测试(左斜树、右斜树、平衡或者不平衡树)
3.特殊测试(一个结点,null)
Java实现
实现一
public class IsBalancedTree {
public static class TreeNode{
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
/**
* 方法一
* @param root
* @return
*/
public boolean IsBalanced_Solution(TreeNode root) {
if (root == null) return true;
if (Math.abs(treeDepth(root.right) - treeDepth(root.left)) <= 1) {
return IsBalanced_Solution(root.left) && IsBalanced_Solution(root.right);
} else {
return false;
}
}
public int treeDepth(TreeNode root) {
if (root == null) {
return 0;
}
int leftDepth = treeDepth(root.left);
int rightDepth = treeDepth(root.right);
return 1 + Math.max(leftDepth, rightDepth);
}
}
实现二
public class IsBalancedTree {
public static class TreeNode{
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
/**
* 方法二
* 利用后续遍历,能够先遍历子节点,记录下深度,再计算父节点是否balance,就能够只遍历一遍。
* @param root
* @return
*/
public boolean IsBalanced_Solution(TreeNode root) {
if (root == null) return true;
return treeDepthPruning(root) != -1;
}
public int treeDepthPruning(TreeNode root) {
if (root == null) {
return 0;
}
int leftDepth = treeDepth(root.left);
if (leftDepth == -1) return -1; // 这是剪枝
int rightDepth = treeDepth(root.right);
if (rightDepth == -1) return -1; // 这是剪枝
if (Math.abs(leftDepth - rightDepth) > 1) {
return -1;
} else {
return 1 + Math.max(leftDepth, rightDepth);
}
}
}
收获
- 判断树不平衡后,可以直接退出(剪枝),不需要遍历完整棵树。