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

最新推荐文章于 2022-06-19 19:26:02 发布

马踏飞燕&lin_li

最新推荐文章于 2022-06-19 19:26:02 发布

阅读量87

点赞数 1

分类专栏：剑指offer

本文链接：https://blog.csdn.net/matafeiyanll/article/details/113699556

版权

剑指offer 专栏收录该内容

27 篇文章 1 订阅

订阅专栏

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

输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。

示例 1:

给定二叉树 [3,9,20,null,null,15,7]

    3
   / \
  9  20
    /  \
   15   7

返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]

       1
      / \
     2   2
    / \
   3   3
  / \
 4   4

返回 false 。

限制：

0 <= 树的结点个数 <= 10000

思路一：

在遍历每一个节点时，调用函数Maxdepth得到它的左，右子树的深度。如果每个节点的左右子树深度相差都不超过1，那么按照定义就是一棵平衡二叉树。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int maxDepth(TreeNode* root) {
        if(root==NULL) return 0;
        else return max(maxDepth(root->left),maxDepth(root->right))+1;
    }
    bool isBalanced(TreeNode* root) {
        if(!root) return true;
        TreeNode* left = root->left;
        TreeNode* right = root->right;
        int diff = maxDepth(left) - maxDepth(right);
        if(abs(diff)>1) return false;
        return isBalanced(left) && isBalanced(right);
    }
};

上述代码固然简洁，但是每一个节点会被重复遍历多次，效率不高。
在这里插入图片描述

思路二：

每个节点只遍历一次解法，用后序遍历的方式遍历二叉树的每个节点，那么在遍历到一个节点之前我们就已经遍历了它的左、右子树。只要在遍历每个节点的时候记录它的深度，我们就可以一边遍历一边判断每个节点是不是平衡的。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isBalanced(TreeNode* root) {
        int depth = 0;
        return isBalanced(root,&depth);
    }
    bool isBalanced(TreeNode* root,int* pDepth){
        if(root==NULL){
            *pDepth = 0;
            return true;
        }
        int left,right;
        if(isBalanced(root->left,&left) && isBalanced(root->right,&right)){
            int diff = left - right;
            if(abs(diff)<=1){
                *pDepth = 1+(left>right?left:right);
                return true;
            }
        }
        return false;
    }
};