【leetcode】110. 平衡二叉树（balanced-binary-tree）（DFS）[简单]

最新推荐文章于 2024-11-10 14:06:08 发布

KroneX

最新推荐文章于 2024-11-10 14:06:08 发布

阅读量168

点赞数

分类专栏： DFS leetcode 题解文章标签： leetcode dfs

本文链接：https://blog.csdn.net/Krone_/article/details/108052395

版权

题解同时被 3 个专栏收录

217 篇文章 0 订阅

订阅专栏

leetcode

215 篇文章 0 订阅

订阅专栏

DFS

23 篇文章 0 订阅

订阅专栏

链接

https://leetcode-cn.com/problems/balanced-binary-tree/

耗时

解题：28 min
题解：7 min

题意

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

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点的左右两个子树的高度差的绝对值不超过1。

思路

dfs() 返回树的高度，在回溯时判断每个节点左右子树的高度差是否大于 1，如果存在一个节点的左右差值的绝对值大于 1，则说明这不是一颗平衡树，ans = false。而如果所有的节点的左右差值均小于等于 1，则说明这是一颗平衡树。

时间复杂度： $O (n o d e)$ ，node 是树的节点数

AC代码

/**
 * 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 ans = true;
    int dfs(TreeNode* root) {
        if(ans == false) return -1;
        if(root == NULL) return 0;
        int l = dfs(root->left);
        int r = dfs(root->right);
        if(abs(r-l) > 1) {
            ans = false;
            return -1;
        }
        return max(l, r) + 1;
    }
    bool isBalanced(TreeNode* root) {
        if(root == NULL) return true;
        dfs(root);
        return ans;
    }
};