平衡二叉树
题目
输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
1 < = 树 的 结 点 个 数 < = 10000 1 <= 树的结点个数 <= 10000 1<=树的结点个数<=10000
思路
对二叉树做后序遍历时,遍历到一个节点之前已经遍历了其左右子树。
所以只要在遍历每个节点时记录其深度,从底至顶返回子树深度,若判定某子树不是平衡树则 “剪枝” ,直接向上返回。
- 当节点root 左 / 右子树的深度差≤1 :则返回当前节点的深度,
- 当节点root 左 / 右子树的深度差 > 2 :则返回 -1,代表此子树不是平衡树 。
时间复杂度
O
(
N
)
O(N)
O(N) : N 为树的节点数;最差情况下,需要递归遍历树的所有节点。
空间复杂度
O
(
N
)
O(N)
O(N): 最差情况下(树退化为链表时),系统递归需要使用 O(N) 的栈空间。
C++
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
bool isBalanced(TreeNode* root) {
if(root==nullptr)
return true;
int depth = 0;
return judge(root, depth);
}
private:
bool judge(TreeNode* root, int &depth)
{
if(root==nullptr)
{
depth = 0;
return true;
}
int left_depth = 0;
int right_depth = 0;
if(judge(root->left, left_depth) && judge(root->right, right_depth)) //先判断左右,都满足了再判断根
{
int diff = left_depth-right_depth;
if(abs(diff)<=1)
{
depth = max(left_depth, right_depth)+1;
return true;
}
}
return false;
}
};
python
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def isBalanced(self, root: TreeNode) -> bool:
def judge(root: TreeNode):
if not root:
return 0
left = judge(root.left)
if left == -1:
return -1
right = judge(root.right)
if right == -1:
return -1
if abs(left-right)<=1:
return max(left, right)+1
else:
return -1
return judge(root) != -1