欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Leetcode:平衡二叉树

程序员文章站 2022-02-19 06:34:06
...

给定一个二叉树,判断它是否是高度平衡的二叉树。

本题中,一棵高度平衡二叉树定义为:

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过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 。

/**
 * 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) {
        return height(root)!=-1;//等于-1时不平衡
    }

    int height(TreeNode *node){
        //空节点返回0
        if(node==NULL) return 0;
        int lc = height(node->left);
        int rc = height(node->right);
        
        if(lc>=0 && rc>=0 && abs(lc-rc)<=1)
            return max(lc,rc)+1;
        else
            return -1;
    }
};