1. 平衡二叉树
描述:
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
1.1 从顶到底
思路:
从顶到底,O(N^2),基础
(1)定义辅助方法depth求得当前树的高度
(2)若root为空,是平衡的,返回true
(3)求左子树和右子树的高度
(4)若左子树和右子树高度差的绝对值不超过1(利用Math.abs()求绝对值)并且左右子树都平衡(递归),返回true
代码:
class Solution {
public boolean isBalanced(TreeNode root) {
if(root == null) return true;
int leftdepth = depth(root.left);
int rightdepth = depth(root.right);
return Math.abs(leftdepth - rightdepth) < 2 && isBalanced(root.left) && isBalanced(root.right);
}
public int depth(TreeNode root){
if(root == null) return 0;
int leftdepth = depth(root.left);
int rightdepth = depth(root.right);
return leftdepth > rightdepth ? leftdepth + 1 : rightdepth + 1;
}
}
1.2 从底到顶
思路:
从底到顶,O(N),进阶
(1)建立辅助方法height求此时平衡树的高度
(2)若root为空,认为平衡,返回true
(3)height方法,返回值为当前root平衡树的高度,若root为空返回0
(4)递归获取左右子树的高度,若任一子树(左右)高度为-1或左右子树差的绝对值>1,则返回-1代表不平衡
否则返回最高平衡子树+1
(5)返回 若root平衡树高度>=0为true
代码:
class Solution {
public boolean isBalanced(TreeNode root) {
if(root == null) return true;
return height(root) >= 0;
}
public int height(TreeNode root){
if(root == null) return 0;
int lefth, righth;
if((lefth = height(root.left)) == -1 || (righth = height(root.right)) == -1 || Math.abs(lefth - righth) > 1) return -1;
return Math.max(lefth, righth) + 1;
}
}