剑指 Offer 55 - II. 平衡二叉树

Amberfd

于 2021-03-04 14:02:53 发布

阅读量79

点赞数

分类专栏：剑指offer 文章标签： c++

本文链接：https://blog.youkuaiyun.com/Amberfd/article/details/114366937

版权

剑指offer 专栏收录该内容

72 篇文章

订阅专栏

在这里插入图片描述

解法一

class Solution {
public:
    bool isBalanced(TreeNode* root) {
        return deepth(root)!=-1;
    }

    int deepth(TreeNode* root){
        if(root==NULL) return 0;
        int left=deepth(root->left);
        if(left==-1) return -1;
        int right=deepth(root->right);
        if(right==-1) return -1;
        return (abs(left-right)<2)? max(left,right)+1:-1;
    }
};

解法二

class Solution {
public:
    bool res=true;
    bool isBalanced(TreeNode* root) {
        dfs(root);
        return res;
    }

    int dfs(TreeNode* root){
        if(root==NULL) return 0;
        int left=dfs(root->left), right=dfs(root->right);
        if(abs(left-right)>1) res=false;
        return max(left,right)+1;
    }
};