PAT甲级 1099 Build A Binary Search Tree

题目描述

题目的意思是给你N个节点，前N行表示每个节点（从0开始）对应的左右节点分别是谁构成一个树，然后给你N个数，将这N个数匹配到每一个节点使这棵树构成一个二叉搜索树。

解题思路

首先我们得知道二叉搜索树的特性，二叉搜索树的特性是该节点的左子树都比它小，右子树都比它大。我们肯定得先构造树，可以借助结构体，分别构建左子树和右子树，然后对给出的N个数进行从小到大的排序。都处理完毕后就要选择节点了，可以用深搜找到最里面的左子树一直往上走，就可以对每一个元素赋值，最后使用队列查找一下各层就是层次遍历的结果。

代码

#include <bits/stdc++.h>
using namespace std;
int n,u,v,b[110],r[110]={0},c,f=0;
vector<int>e[110];
struct node
{
	int lchild=-1;
	int rchild=-1;
}a[110];//使用结构体构建树
void dfs(int root)//深搜
{
	if(a[root].lchild!=-1)
	{
		dfs(a[root].lchild);
	}
	r[root]=b[f];//找到最里面，f表示第几个大的节点
	f++;
	if(a[root].rchild!=-1)
	{
		dfs(a[root].rchild);
	}
}
queue<int>q;
void level()//层次遍历
{
	while(!q.empty())//借助队列
	{
		int t=q.front();
		q.pop();
		if(a[t].lchild!=-1)
		{
			cout<<" "<<r[a[t].lchild];
			q.push(a[t].lchild);
		}
		if(a[t].rchild!=-1)
		{
			cout<<" "<<r[a[t].rchild];
			q.push(a[t].rchild);
		}
	}
	
}
int main()
{
	cin>>n;
	for(int i=0;i<n;i++)
	{
		cin>>u>>v;
		if(u!=-1)
		{
			a[i].lchild=u;
			r[u]++;
			}	
		if(v!=-1)
		{
			r[v]++;//找到没有入度的节点，就是根节点
			a[i].rchild=v;
			}	
	}
	for(int i=0;i<n;i++)
	{
		cin>>b[i];
	}
	sort(b,b+n);
	for(int i=0;i<n;i++)
	{
		if(r[i]==0)
		{
			c=i;
			break;
		}
		r[i]=0;
	}
	dfs(c);
	q.push(c);
	cout<<r[c];
	level();
	return 0;
 }