代码随想录 二叉树—平衡二叉树
思路:平衡二叉树的特点是每个节点的左右子树的高度差的绝对值不超过1。小tips:与高度相关的用后序遍历,与深度相关的用前序遍历。高度是自下而上来算,深度是自上而下像钻井一般。那这里用后序遍历,左右中。先处理左子树,这里的-2作为信号,换成别的负数也没问题。如果左子树就不是平衡二叉树,那就继续把-2信号返回,再处理右子树同理。最后处理中间,判断左右子树的高度差,如果大于1,就继续返回-2信号,如果小于等于1就返回父节点的高度,即1+左右子树较大值。
题解c++:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int getHeight(TreeNode* node)
{
if(node==NULL) return 0;
int leftHeight=getHeight(node->left);
if(leftHeight==-2) return -2;
int rightHeight=getHeight(node->right);
if(rightHeight==-2) return -2;
return abs(leftHeight-rightHeight)>1 ? -2:1+max(leftHeight,rightHeight);
}
bool isBalanced(TreeNode* root) {
return getHeight(root)== -2 ? false:true;
}
};