平衡二叉树
题目描述
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
思路
- 在计算二叉树的深度的基础上加以判断是否是平衡二叉树
- 用后序遍历,在遍历一个节点之前我们已经遍历了他的左右子树,在遍历每个节点的时候记录他的深度,当该子树是平衡二叉树的时候我们返回它的深度,若不是则返回-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;
}
}
时间: 2024-10-13 12:37:21