170.二叉树:均衡二叉树(力扣)
https://img-blog.csdnimg.cn/direct/c7eef1902cad478ab431b8ae1df2a4cc.png代码解决
/**
* 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 {
public:
// 递归计算二叉树的高度,同时检查是否平衡
int height(TreeNode* root) {
// 基本条件:如果节点为空,高度为0
if (root == nullptr) return 0;
// 递归计算左子树的高度
int leftdeep = height(root->left);
// 如果左子树不平衡,返回-1
if (leftdeep == -1) return -1;
// 递归计算右子树的高度
int rightdeep = height(root->right);
// 如果右子树不平衡,返回-1
if (rightdeep == -1) return -1;
// 如果左右子树高度差大于1,表示当前树不平衡,返回-1
if (abs(leftdeep - rightdeep) > 1) return -1;
// 否则返回当前树的高度,即左右子树高度的最大值加1
return 1 + max(leftdeep, rightdeep);
}
// 判断二叉树是否平衡
bool isBalanced(TreeNode* root) {
// 调用height方法,如果返回值为-1,表示树不平衡,返回false
return height(root) == -1 ? false : true;
}
};
[*] TreeNode 结构体定义:
[*]val:节点的整数值。
[*]left:指向左子节点的指针。
[*]right:指向右子节点的指针。
[*] Solution 类:
[*]包含两个方法:height 和 isBalanced。
[*] height 方法:
[*]这是一个递归函数,用来盘算二叉树的高度,并查抄树是否均衡。
[*]根本条件:如果 root 为 nullptr,则返回高度 0。
[*]递归盘算左子树和右子树的高度:
[*]如果左子树的高度为 -1,表现左子树不均衡,直接返回 -1。
[*]如果右子树的高度为 -1,表现右子树不均衡,直接返回 -1。
[*]如果左右子树的高度差超过 1,表现当前树不均衡,返回 -1。
[*]否则,返回当前树的高度,即左右子树高度的最大值加 1。
[*] isBalanced 方法:
[*]这个方法调用 height 方法,如果返回值为 -1,表现树不均衡,返回 false。否则,返回 true,表现树是均衡的。
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页:
[1]