110 平衡二叉树
概念普及
简单路径:路径上的各顶点均不互相重复
二叉树节点的深度:指从根节点到该节点的最长简单路径边的条数
二叉树节点的高度:指从该节点到叶子节点的最长简单路径边的条数
求深度可以从上到下去查,所以需要前序遍历,而高度只能从小到上去查,所以只能后序遍历
题目
解题思路
- 明确递归函数的参数和返回值:参数:当前传入的节点, 返回值:当前传入节点距离根节点的高度
- 明确终止条件:传入节点为null,返回0
- 明确单层递归的逻辑:判断以当前传入节点为根节点的二叉树是否为平衡二叉树,即左右子树高度差是否超过1,如果超过1,则返回-1(以下一次回溯,表示该二叉树已经不是平衡二叉树),如果小于1,则返回左右子树最大者+1
- 明确递归三步曲,自底向上递归,时间复杂度为O(n),每个节点仅访问依次,空间复杂度O(n),取决于递归调用层数,递归调用层数不会超过n
public boolean isBalanced(TreeNode root) {
return count(root) >= 0;
}
public int count(TreeNode root){
if(root == null) return 0;
int leftLength = count(root.left);
int rightLength = count(root.right);
if(leftLength == -1 || rightLength == -1 || Math.abs(leftLength-rightLength) > 1) return -1;
return Math.max(leftLength, rightLength) + 1;
}