leetcode平衡二叉树
题目:输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
链接:https://leetcode-cn.com/problems/balanced-binary-tree/
题解:
解法1,自顶向下,先判断左右子树是否平衡,再判断当前树是否平衡
public boolean isBalanced(TreeNode root) {
if(root == null)
return true;
return isBalanced(root.left) && isBalanced(root.right) && Math.abs(maxDepth(root.left) - maxDepth(root.right)) <= 1;
}
public int maxDepth(TreeNode root) {
if(root == null)
return 0;
return Math.max(maxDepth(root.left), maxDepth(root.right)) + 1;
}
解法二,在上面的题解中会重复获取子树的高度,因为对子树调用isBalanced时只获得了该子树是否是平衡树的信息,子树的高度却并未返回。可以使用数组在返回是否是平衡树信息的时候同时返回该子树的高度,但我们容易观察到树的高度是正值,因此可以用负数表示该数不是平衡子树,于是可得到以下代码。
public boolean isBalanced(TreeNode root) {
if(root == null || maxDepth(root) != -1)
return true;
else
return false;
}
public int maxDepth(TreeNode root) {
if(root == null)
return 0;
int lenLeft = maxDepth(root.left);
int lenRight = maxDepth(root.right);
if(lenLeft == -1 || lenRight == -1 || Math.abs(lenLeft - lenRight) > 1)
return -1;
else
return Math.max(lenLeft, lenRight) + 1;
}