【刷题之路】平衡二叉树

输入一棵二叉树,判断该二叉树是否是平衡二叉树。

平衡二叉树的定义:左右子树节点差不大于1,并且左右子树也都是平衡二叉树

实际上与计算二叉树深度异曲同工,对于每一个节点,需要返回左右子树的最大深度,并且比较差值,并返回左右子树的最大值作为当前节点的深度

代码如下:

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
int dep = 0;
        return IsBalanced(pRoot, dep);
    }
    bool IsBalanced(TreeNode* root ,int &dep){  //需要将每一层的深度不断传递
        if(root==NULL) return true;
        int left=0,right=0;
        if(IsBalanced(root->left,left) && IsBalanced(root->right,right)){  //判断条件
            int dif = left – right; 
            if(dif<-1 || dif >1)  //如果差值不满足1之内,则直接返回false
                return false;
            dep = (left > right ? left : right) + 1; //如果满足,则返回左右子树较大值作为当前节点深度,参与上一节点比较
            return true;
        }
        return false;
    }
};

    原文作者:平衡二叉树
    原文地址: https://blog.csdn.net/zyn2609530/article/details/52053550
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞