二叉搜索树 (BST) 递归定义为具有以下属性的二叉树:
- 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值
- 若它的右子树不空,则右子树上所有结点的值均大于或等于它的根结点的值
- 它的左、右子树也分别为二叉搜索树
给定二叉树的具体结构以及一系列不同的整数,只有一种方法可以将这些数填充到树中,以使结果树满足 BST 的定义。
请你输出结果树的层序遍历。
示例如图 11 和图 22 所示。
输入格式
第一行包含一个正整数 NN,表示树的结点个数。
所有结点的编号为 0∼N−10∼N−1,并且编号为 00 的结点是根结点。
接下来 NN 行,第 ii 行(从 00 计数)包含结点 ii 的左右子结点编号。如果该结点的某个子结点不存在,则用 −1−1 表示。
最后一行,包含 NN 个不同的整数,表示要插入树中的数值。
输出格式
输出结果树的层序遍历序列。
数据范围
1≤N≤1001≤N≤100
输入样例:
9
1 6
2 3
-1 -1
-1 4
5 -1
-1 -1
7 -1
-1 8
-1 -1
73 45 11 58 82 25 67 38 42
输出样例:
58 25 82 11 38 67 45 73 42
#include <iostream>
#include <queue>
#include <algorithm>
using namespace std;
const int N=110;
queue<int>q;
int w[N],tr[N],k,l[N],r[N];
void dfs(int u)
{
if(u==-1) return ;
if(l[u]!=-1) dfs(l[u]);
tr[u]=w[k++];
if(r[u]!=-1) dfs(r[u]);
}
void bfs()
{
q.push(0);
int flag=1;
while(!q.empty())
{
auto t=q.front();
if(!flag) cout<<" ";
flag=0;
cout<<tr[t];
q.pop();
if(l[t]!=-1) q.push(l[t]);
if(r[t]!=-1) q.push(r[t]);
}
}
int main()
{
int n;
cin>>n;
for(int i=0;i<n;i++) cin>>l[i]>>r[i];
for(int i=0;i<n;i++) cin>>w[i];
sort(w,w+n);
dfs(0);
bfs();
}