LeetCode - 110. 平衡二叉树

描述

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

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

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

 

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：true
示例 2：

输入：root = [1,2,2,3,3,null,null,4,4]
输出：false
示例 3：

输入：root = []
输出：true
 

提示：

树中的节点数在范围 [0, 5000] 内
-104 <= Node.val <= 104

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/balanced-binary-tree/

求解

   class Solution {
    private:
        std::unordered_map<TreeNode*, int> record; // 记忆某个节点为根节点的二叉树高度，避免重复计算
        // 递归 + 记忆化搜索求解二叉树高度
        int depth(TreeNode* root) {
            if(!root){
                return 0;
            }
            if(record.count(root) >0) {
                return record[root];
            } else {
                return std::max(depth(root->left), depth(root->right))  + 1;
            }
        }
    public:
        // 迭代求解，层序遍历每个节点判断以其为根节点的二叉树是否平衡
        bool isBalanced_useLeverOrder(TreeNode* root) {
            if(!root) {
                return true;
            }

            std::queue<TreeNode*> q;
            q.push(root);
            while(!q.empty()) {
                auto node =  q.front();
                q.pop();

                if(std::abs(depth(node->left) - depth(node->right)) > 1) {
                    return false;
                }

                if(node->left) {
                    q.push(node->left);
                }
                if(node->right) {
                    q.push(node->right);
                }
            }
            return true;
        }

        // 递归求解，深度遍历（前序）每个节点判断以其为根节点的二叉树是否平衡
        bool isBalanced(TreeNode* root) {
            if(!root){
                return true;
            }
            if(std::abs(depth(root->left) - depth(root->right)) <= 1 ) {
                return isBalanced(root->left) && isBalanced(root->right);
            } else {
                return false;
            }
        }
    };