平衡二叉树
题目描述
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
思路
- 在计算二叉树的深度的基础上加以判断是否是平衡二叉树
- 用后序遍历,在遍历一个节点之前我们已经遍历了他的左右子树,在遍历每个节点的时候记录他的深度,当该子树是平衡二叉树的时候我们返回它的深度,若不是则返回-1
代码
public class Solution {
public boolean IsBalanced_Solution(TreeNode root) {
boolean result = false; if (root == null) { return true; } if (isBalanceTree(root) > 0) { result = true; } return result; } public int isBalanceTree(TreeNode root) { if (root == null) { return 0; } int left = isBalanceTree(root.left); int right = isBalanceTree(root.right); if (left >= 0 && right >= 0) { int diff = left - right; if (diff >= -1 && diff <= 1) { int result = 1 + ((left > right) ? left: right); return result; } } return -1; } }