C++中检查二叉树的完整性


假设我们有一个二叉树。我们必须检查这棵树是否是完整的二叉树。n级的完整二叉树具有n-1个完整的级别,并且n级的所有节点都从左侧填充。所以如果输入树是这样的:


那么输出将为真,因为这是一个完整的二叉树。

为了解决这个问题,我们将遵循以下步骤:

  • 如果树为空,则返回null

  • 创建一个队列q并将根节点插入其中

  • 设置flag := true

  • 当q中有一些元素时

    • sz := 队列的大小

    • 当sz不为0时

      • node := 从队列中删除后的节点

      • 如果节点有左子树,则

        • 如果flag已设置,则将节点的左子树插入q中,否则返回false

      • 否则flag := false

      • 如果节点有右子树,则

        • 如果flag已设置,则将节点的右子树插入q中,否则返回false

      • flag := false

      • sz := sz – 1

  • 返回true

让我们看看下面的实现以更好地理解:

示例

 在线演示

#include <bits/stdc++.h>
using namespace std;
class TreeNode{
   public:
   int val;
   TreeNode *left, *right;
   TreeNode(int data){
      val = data;
      left = NULL;
      right = NULL;
   }
};
void insert(TreeNode **root, int val){
   queue<TreeNode*> q;
   q.push(*root);
   while(q.size()){
      TreeNode *temp = q.front();
      q.pop();
      if(!temp->left){
         if(val != NULL)
            temp->left = new TreeNode(val);
         else
            temp->left = new TreeNode(0);
         return;
      }else{
         q.push(temp->left);
      }
      if(!temp->right){
         if(val != NULL)
            temp->right = new TreeNode(val);
         else
            temp->right = new TreeNode(0);
         return;
      }else{
         q.push(temp->right);
      }
   }
}
TreeNode *make_tree(vector<int> v){
   TreeNode *root = new TreeNode(v[0]);
   for(int i = 1; i<v.size(); i++){
      insert(&root, v[i]);
   }
   return root;
}
class Solution {
   public:
   bool isCompleteTree(TreeNode* root) {
      if(!root)return true;
      queue <TreeNode*> q;
      q.push(root);
      bool isComplete = true;
      while(!q.empty()){
         int sz = q.size();
         while(sz--){
            TreeNode* node = q.front();
            q.pop();
            if(node->left){
               if(isComplete){
                  q.push(node->left);
               }else return false;
            }else{
               isComplete = false;
            }
            if(node->right){
               if(isComplete){
                  q.push(node->right);
               }else return false;
            }else{
               isComplete = false;
            }
         }  
      }
      return true;
   }
};
main(){
   vector<int> v = {1,2,3,4,5,6};
   TreeNode *r1 = make_tree(v);
   Solution ob;
   cout << (ob.isCompleteTree(r1));
}

输入

{1,2,3,4,5,6}

输出

1

更新于:2020年5月2日

172 次浏览

开启您的职业生涯

完成课程后获得认证

开始学习
广告