平衡二叉树

最新推荐文章于 2019-03-13 16:34:21 发布

ningbo2016

最新推荐文章于 2019-03-13 16:34:21 发布

阅读量131

点赞数

分类专栏：面试编程题文章标签：二叉树

本文链接：https://blog.csdn.net/ningbo2016/article/details/84177468

版权

面试编程题专栏收录该内容

32 篇文章 0 订阅

订阅专栏

题目描述

输入一棵二叉树，判断该二叉树是否是平衡二叉树。
思路：平衡二叉树是指左右子树的深度值不能超过1，递归，对每个结点遍历获取深度，进行比较，这里有2种想法，一种从上到下进行比较，另一种是从下往上进行比较，从上到下进行比较，会出现结点多次遍历的情况。
代码（从上到下）：

class Solution {
public:
    int depth(TreeNode* root)
    {
        if(root==NULL) return 0;
        return 1+max(depth(root->left),depth(root->right));
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(pRoot==NULL) return true;
        return abs(depth(pRoot->left)-depth(pRoot->right))<=1&&
            IsBalanced_Solution(pRoot->left)&&
            IsBalanced_Solution(pRoot->right);
    }
};

代码（从下往上）：//后续递归遍历

class Solution {
public:
    int depth(TreeNode* root)
    {
        if(root==NULL) return 0;
        int left = depth(root->left);
        if(left==-1) return -1;
        int right = depth(root->right);
        if(right==-1) return -1;
        return abs(left - right) > 1 ? -1 : 1 + max(left, right);
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        return depth(pRoot)!=-1;
    }
};