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
广告