170.二叉树:均衡二叉树(力扣)

打印 上一主题 下一主题

主题 656|帖子 656|积分 1968


代码解决

  
  1. /**
  2. * Definition for a binary tree node.
  3. * struct TreeNode {
  4. *     int val;
  5. *     TreeNode *left;
  6. *     TreeNode *right;
  7. *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
  8. *     TreeNode(int x) : val(x), left(nullptr, right(nullptr) {}
  9. *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
  10. * };
  11. */
  12. class Solution {
  13. public:
  14.     // 递归计算二叉树的高度,同时检查是否平衡
  15.     int height(TreeNode* root) {
  16.         // 基本条件:如果节点为空,高度为0
  17.         if (root == nullptr) return 0;
  18.         // 递归计算左子树的高度
  19.         int leftdeep = height(root->left);
  20.         // 如果左子树不平衡,返回-1
  21.         if (leftdeep == -1) return -1;
  22.         // 递归计算右子树的高度
  23.         int rightdeep = height(root->right);
  24.         // 如果右子树不平衡,返回-1
  25.         if (rightdeep == -1) return -1;
  26.         // 如果左右子树高度差大于1,表示当前树不平衡,返回-1
  27.         if (abs(leftdeep - rightdeep) > 1) return -1;
  28.         // 否则返回当前树的高度,即左右子树高度的最大值加1
  29.         return 1 + max(leftdeep, rightdeep);
  30.     }
  31.    
  32.     // 判断二叉树是否平衡
  33.     bool isBalanced(TreeNode* root) {
  34.         // 调用height方法,如果返回值为-1,表示树不平衡,返回false
  35.         return height(root) == -1 ? false : true;
  36.     }
  37. };
复制代码


  • 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企服之家,中国第一个企服评测及商务社交产业平台。
回复

使用道具 举报

0 个回复

正序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

干翻全岛蛙蛙

金牌会员
这个人很懒什么都没写!

标签云

快速回复 返回顶部 返回列表