递归的方式判断,因为是判断高度,因此我们需要从底至上进行判断。左右子树如果高度差大于1则说明不平衡。
class Solution {
public boolean isBalanced(TreeNode root) {
return getHeight(root) != -1;
}
private int getHeight(TreeNode root) {
if (root == null) {
return 0;
}
int leftHeight = getHeight(root.left);
if (leftHeight == -1) {
return -1;
}
int rightHeight = getHeight(root.right);
if (rightHeight == -1) {
return -1;
}
// 左右子树高度差大于1,return -1表示已经不是平衡树了
if (Math.abs(leftHeight - rightHeight) > 1) {
return -1;
}
return Math.max(leftHeight, rightHeight) + 1;
}
}
这里用迭代法,进行二叉树的层序遍历,每次遍历记录总节点数+1
这里补充一些队列的常用接口:
class Solution {
public int countNodes(TreeNode root) {
if (root == null) return 0;
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
int result = 0;
while (!queue.isEmpty()) {
int size = queue.size();
while (size -- > 0) {
TreeNode cur = queue.poll();
result++;
if (cur.left != null) queue.add(cur.left);
if (cur.right != null) queue.add(cur.right);
}
}
return result;
}
}
//递归判断节点,最后返回深度时记得+1.
class Solution {
public int maxDepth(TreeNode root) {
if(root==null)return 0;
int left = maxDepth(root.left);
int right = maxDepth(root.right);
return Math.max(left,right)+1;
}
}
这题与上题不同,这个要先判断是否为空节点,然后判断是否抵达了叶子节点,递归的方式获得左右子树的最大深度,然后取min就是所求,
如果都不为空,那么就取最小值,如果都是空那么就返回1
如果左右子节点又一个是为null的话,那就m1+m2+1 因为有个为空所以 m1,m2有个必是0.
class Solution {
public int minDepth(TreeNode root) {
if(root==null)return 0;
if(root.left==null&&root.right==null)return 1;
int m1=minDepth(root.left);
int m2=minDepth(root.right);
if(root.left==null||root.right==null)return m1+m2+1;
return Math.min(m1,m2)+1;
}
}