剑指OFFER 平衡二叉树

Posted 2021-03-11 virgildevil

剑指OFFER 平衡二叉树

分析

先理解什么是平衡二叉树

它是一棵空树或它的左右两个子树的高度差的绝对值不超过1，并且左右两个子树都是一棵平衡二叉树。

下面的代码我只判断了根节点左右孩子的深度(没有再递归判断其子树的),但是也一样过了.

代码

class Solution {
public:
    
    int left_depth = 0;
    int right_depth = 0;
    int* p_depth = NULL;
    
    void recur(TreeNode* node,int depth)
    {
        if(node == NULL)return ;
        
        if(depth>*p_depth)*p_depth=depth;
        
        recur(node->left,depth+1);
        recur(node->right,depth+1);
    }
    
    bool IsBalanced_Solution(TreeNode* root) {
        if(root == NULL)return true;
        p_depth = &left_depth;
        recur(root->left,1);
        p_depth = &right_depth;
        recur(root->right,1);
        return !(abs(left_depth-right_depth)>1);
    }
};