Написано: 16.03.2023
110. Сбалансированное двоичное дерево (Balanced Binary Tree)
Задание.
Получено двоичное дерево.
Определите, является ли оно сбалансированным-по-высоте.
Сбалансированное-по-высоте двоичное дерево – это двоичное дерево, в котором глубина двух поддеревьев каждого узла никогда не отличается больше, чем на единицу.
Пример 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
Решение.
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
private:
bool ans;
int checkBalance(TreeNode* root){
if(root == NULL)
return 0;
if(ans == false)
return 0;
int leftTree = checkBalance(root->left);
int rightTree = checkBalance(root->right);
if(abs(leftTree - rightTree) > 1) {
ans = false;
}
return 1 + max(leftTree, rightTree);
}
public:
bool isBalanced(TreeNode* root){
ans = true;
checkBalance(root);
return ans;
}
};