平衡二叉树的定义是左右子树高度差不超过1,同时左右子树也是平衡二叉树,于是代码逻辑可以如下
1.判断树是否为空,空则返回true
2.判断左右子树深度差,其中,求树深度的函数在上一题中“二叉树的深度中”已实现,差超过1,返回false
3.若通过2的判断,对左右子树也判断是否都是平衡二叉树,判断函数为函数自身,递归调用
public class Solution { public int TreeDepth(TreeNode root) { if(root==null){ return 0; } if(root.left==null && root.right==null){ return 1; } return 1+Math.max(TreeDepth(root.left),TreeDepth(root.right)); } public boolean IsBalanced_Solution(TreeNode root) { if(root==null) return true; if(Math.abs(TreeDepth(root.left)-TreeDepth(root.right))>1) return false; return IsBalanced_Solution(root.left) && IsBalanced_Solution(root.right); } }