二叉树平衡检查

题目:实现一个函数,检查二叉树是否平衡,平衡的定义如下,对于树中的任意一个结点,其两颗子树的高度差不超过1。给定指向树根结点的指针TreeNode* root,请返回一个bool,代表这棵树是否平衡。

思路:根据题目定义,子树的高度差超过1即为不平衡,所以可以写一个求子树高度的函数,用递归实现比较方便

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/

class Balance {
public:
    int getHeight(TreeNode* node){
        if(node==NULL)
            return 0;
        else
            return getHeight(node->left)>getHeight(node->right)?getHeight(node->left)+1:getHeight(node->right)+1;
    }
    
    bool isBalance(TreeNode* root) {
        // write code here
        if(root==NULL)
            return true;
        int heightDiff = getHeight(root->left)-getHeight(root->right);
        if(heightDiff<-1 || heightDiff>1)
            return false;
        else
            return (isBalance(root->left) && isBalance(root->right));
    }
};

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