PAT【L3-010】是否完全二叉搜索树（详解）

原题：

将一系列给定数字顺序插入一个初始为空的二叉搜索树（定义为左子树键值大，右子树键值小），你需要判断最后的树是否一棵完全二叉树，并且给出其层序遍历的结果。

输入格式：

输入第一行给出一个不超过20的正整数N；第二行给出N个互不相同的正整数，其间以空格分隔。

输出格式：

将输入的N个正整数顺序插入一个初始为空的二叉搜索树。在第一行中输出结果树的层序遍历结果，数字间以1个空格分隔，行的首尾不得有多余空格。第二行输出YES，如果该树是完全二叉树；否则输出NO。

输入样例1：

9
38 45 42 24 58 30 67 12 51

输出样例1：

38 45 24 58 42 30 12 67 51
YES

输入样例2：

8
38 24 12 45 58 67 42 51

输出样例2：

38 45 24 58 42 12 67 51
NO

思路：

首先，根据题意，构建一棵二叉搜索树。创建完之后，通过队列对它进行bfs广度优先遍历（层序遍历)。

建立好二叉搜索树以后的难点应该就是判断是否是完全二叉树了。完全二叉树的定义是：深度为n的二叉树，其全部深度n-1的节点为满，而深度为n的节点只能聚集在左侧。那么我们可以层次遍历找到第一个NULL节点，记录遍历的节点个数，如果其个数为n的话说明为完全二叉树，否则不是。

百度里的定义是：完全二叉树是效率很高的数据结构，完全二叉树是由满二叉树而引出来的。对于深度为K的，有n个结点的二叉树，当且仅当其每一个结点都与深度为K的满二叉树中编号从1至n的结点一一对应时称之为完全二叉树。

代码：

#include<bits/stdc++.h>
using namespace std;
int n;
struct treeNode
{
	int data;
	treeNode* left;
	treeNode* right;
};

void insert_tree(treeNode*& node, int data)
{
	if (node == nullptr)
	{
		node = new treeNode();
		//初始化结点
		node->left = node->right = nullptr;
		node->data = data;
	}
	//左子树键值大，右子树键值小
	else if (data > node->data)
	{
		insert_tree(node->left, data);
	}
	else if (data < node->data)
	{
		insert_tree(node->right, data);
	}
}

void init_tree(treeNode*& node, int n)
{
	int dat;
	for (int i = 0; i < n; ++i)
	{
		cin >> dat;
		insert_tree(node, dat);
	}
}
// 判断是否是完全二叉树
bool judge(treeNode* node)
{
	queue<treeNode*>que;
	que.push(node);
	int num = 0;
	while ((node = que.front()) != nullptr)
	{
		que.push(node->left);
		que.push(node->right);
		que.pop();
		num++;
	}
	if (num == n)
		return true;
	return false;
}

// 层序遍历
void view_tree(treeNode* node)
{
	if (node == nullptr)	return;
	queue<treeNode*>que;
	//vector<vector<int>>result;
	que.push(node);
	vector<int>vec;
	while (!que.empty())
	{
		int size = que.size();
		//vector<int>vec;
		for (int i = 0; i < size; ++i)
		{
			treeNode* root = que.front();
			que.pop();
			if (root->data) vec.push_back(root->data);
			if (root->left)	que.push(root->left);
			if (root->right)que.push(root->right);
		}
		//result.push_back(vec);
	}
    cout<<vec[0];
	for (int i = 1; i < vec.size(); i++)
		cout << " "<<vec[i];
}
int main()
{
	
	cin >> n;
	treeNode* root = nullptr;
	//root->left = root->right = nullptr;
	int dat;
	for (int i = 0; i < n; ++i)
	{
		cin >> dat;
		insert_tree(root, dat);
	}
	//init_tree(root, n);
	view_tree(root);
   
	if (judge(root))
		cout << endl<<"YES" << endl;
	else
		cout << endl<<"NO" << endl;

	return 0;
}