C++ 二叉树平衡检查

/**
关于二叉树与图
1.从无到有实现二叉树和图
2.在二叉树和图中检索/查找target value

二叉搜索树:binary search tree
所要满足的条件:对于二叉搜索树中任意一个节点
它的左子树中的所有数值都比当前节点的关键码小
它的右子树中的所有节点的关键码都比当前节点的关键码大

二叉搜索树的结构在一定程度上能够提高搜索效率

平衡二叉树是指:
对于二叉树中的任意一个节点,
节点的左子树的高度与右子树的高度之差不会超过1

二叉树的层数:
根节点的层数为1,从根节点到叶子结点层数依次增加

二叉树的高度:
叶子节点的高度为1,从叶子节点到根节点高度依次增加,根节点的高度最大

二叉树的深度:
叶子节点的深度指的是从根节点到当前叶子节点所经过的路径长度

二叉树的遍历:
熟练实现二叉树的中序遍历,先序遍历和后序遍历
尤其是二叉树的中序遍历:先中序遍历左子树,再访问根节点,最后中序遍历右子树
二叉树遍历的递归实现和迭代实现(尾递归很容易改写成迭代形式)

图的遍历
图的深度优先搜索(DFS)和广度优先搜索(BFS)
二叉树的前序遍历、中序遍历和后序遍历的过程就是一种深度优先搜索

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

算法1:
实现辅助函数 getHeight 以获取层序遍历意义下每个节点的高度
对于二叉树中的任何一个节点,判断其左子树和右子树的高度之差
是否大于1,如果大于1,则立即返回说明该二叉树不是平衡二叉树
但是访问二叉树中的每个节点时，都要递归得到节点的高度,时间开销大

算法2:
构造checkHeight辅助函数,如果输入的节点的左右子树的高度之差大于1,
则直接返回-1表明该二叉树不是平衡二叉树,否则返回当前节点的高度,
以供之后计算它的父代节点的高度做准备


**/
#include<vector>
#include<iostream>
#include<math.h>
using namespace std;

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* root){
    // 得到以当前root节点为根节点的子树的最大高度
    // 使用递归算法实现
        if(root==NULL){
            return 0;
        }
        if(root->left==NULL and root->right==NULL){
            return 1;
        }
        else if(root->left==NULL){
            return 1+this->getHeight(root->right);
        }
        else if(root->right==NULL){
            return 1+this->getHeight(root->left);
        }
        else{
            return 1+max(this->getHeight(root->left),this->getHeight(root->right));
        }
    }

    bool isBalance(TreeNode* root) {
        // write code here
        // 对二叉树进行层序遍历,遍历过程中看到的每个节点,得到它的左子树的高度和右子树的高度
        // 用队列完成对二叉树的层序遍历
        if(root==NULL){
            return true;
        }
        if(root->left==NULL and root->right==NULL){
            return true;
        }
        vector<TreeNode*> helper_queue;// 构造辅助队列以实现对二叉树的层序遍历
        helper_queue.push_back(root);
        TreeNode *temp;
        int left_depth=0;
        int right_depth=0;
        while(helper_queue.size()>0){
            temp=helper_queue[0];
            helper_queue.erase(helper_queue.begin());
            left_depth=this->getHeight(temp->left);
            right_depth=this->getHeight(temp->right);
            if(left_depth-right_depth>1 or right_depth-left_depth>1){
                return false;
            }
            else{
                if(temp->left!=NULL){
                    helper_queue.push_back(temp->left);
                }
                if(temp->right!=NULL){
                    helper_queue.push_back(temp->right);
                }
            }
        }
        return true;
    }
};

int main(){
    return 0;
}


// 算法2
#include<vector>
#include<iostream>
#include<math.h>
using namespace std;

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

class Balance {
public:
    int checkHeight(TreeNode* root){
    // 检查以root节点为根节点的子树是否是平衡二叉树
    // 如果以当前节点为根节点的子树是平衡二叉树，则返回
    // 当前节点的高度，否则，返回-1
        if(root==NULL){
            return 0;
        }
        if(root->left==NULL and root->right==NULL){
            return 1;
        }
        else{
            // 说明当前节点的左右节点都不为空
            int left_height=this->checkHeight(root->left);
            if(left_height==-1){
                return -1;
            }
            int right_height=this->checkHeight(root->right);
            if(right_height==-1){
                return -1;
            }
            if(left_height-right_height>1 or left_height-right_height<-1){
                return -1;
            }
            else{
                return 1+max(left_height,right_height);
            }
        }
    }
    bool isBalance(TreeNode* root) {
        // write code here
        if(this->checkHeight(root)==-1){
            // 只需要将root根节点输入到checkHeight函数中，检查
            // 以root为根节点的左子树和右子树的高度之差是否大于1
            return false;
        }
        else{
            return true;
        }
    }
};