平衡二叉树:它是一棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。
二叉排序树:对于每一个节点,当前节点大于左子树的所有节点,小于右子树的所有节点
要判断是否为二叉树需要满足两个条件:
1.二叉树是否为二叉排序树
2.对于二叉树的每个节点,它的左右两个子树的高度差的绝对值是否<1
牛客这道题已经默认是二叉排序树了,所以不用进行条件1的判断
public class Solution { private boolean ans; public boolean IsBalanced_Solution(TreeNode root) { if(root==null){ return true; } ans=true;//标记左右子树的高度差是否为1 TreeDepth(root); return ans; } private int TreeDepth(TreeNode node){ if(node==null){ return 0; } if(ans) {//ans=true //当前节点左子树高度 int leftDepth = TreeDepth(node.left); //当前节点右子树高度 int rightDepth = TreeDepth(node.right); if (Math.abs(leftDepth - rightDepth) > 1) { ans = false; } //返回当前节点距离叶子节点的长度(方便其它节点计算左右子树的高度) return Math.max(leftDepth,rightDepth)+1; } return 0;//这个地方返回什么已经不重要了,因为我们已经找到一个节点不满足条件了 } }