c++实现递归和非递归判断一颗二叉树是否是平衡二叉树

最新推荐文章于 2024-08-21 18:16:36 发布

孙仲谋111

最新推荐文章于 2024-08-21 18:16:36 发布

阅读量71

点赞数

文章标签： c++ java 算法

本文链接：https://blog.csdn.net/SUNZHONGWU5234/article/details/131976538

版权

该文介绍了如何使用C++编程来判断一棵二叉树是否为平衡二叉树。提供了两种方法，一种是递归方式，通过计算左右子树的高度来确定；另一种是非递归方式，利用队列进行广度优先搜索。如果左右子树的高度差超过1，则树不平衡，否则平衡。

摘要由CSDN通过智能技术生成

c++实现递归和非递归判断一颗二叉树是否是平衡二叉树

struct TreeNode {
  int val;
  TreeNode *left;
  TreeNode *right;
  TreeNode(int v):val(v),left(nullptr), right(nullptr) {}
}

// 递归形式
int getHeight(TreeNode *root) {
  if(!root) return 0;
  int left = getHeight(root->left);
  int right = getHeight(root—>right);
  return left > right ? left+1 :right+1;
}

bool isBalanced(TreeNode *root) {
  if(!root) return true;
  int leftDep = getHeight(root->left);
  int rightDep = getHeight(root->right);
  if(abs(leftDep - rightDep) > 1) {
    return false;
  }
  return isBalanced(root->left) && isBalanced(root->right);
}


// 非递归形式

bool isBalanced(TreeNode *root) {
  if(!root) return false;
   queue<TreeNode *> que;
   que.push(root);
   while(!que.empty()) {
    TreeNode *tempNode = que.front();
    que.pop();
    int leftDepth = getHeight(tempNode->left);
    int rightDepth = getHeight(tempNode->right);
    if(abs(leftDepth - rightDepth) > 1) {
      return false;
    }
    if(tempNode->left) que.push(tempNode->left);
    if(tempNode->right) que.push(tempNode->right);
   }
   return true;

}