【题解】【AcWing】1589. 构建二叉搜索树

76 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

1589. 构建二叉搜索树

原题传送:AcWing 1589. 构建二叉搜索树

二叉搜索树(BST)递归定义为具有以下属性的二叉树:

  • 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值
  • 若它的右子树不空,则右子树上所有结点的值均大于或等于它的根结点的值
  • 它的左、右子树也分别为二叉搜索树

给定二叉树的具体结构以及一系列不同的整数,只有一种方法可以将这些数填充到树中,以使结果树满足BST的定义。

请你输出结果树的层序遍历。

示例如图 11 和图 22 所示。

输入格式

第一行包含一个正整数 NN ,表示树的结点个数。

所有结点的编号为 0N10 \sim N-1 ,并且编号为 00 的结点是根结点。

接下来 NN 行,第 ii 行(从 00 计数)包含结点 ii 的左右子结点编号。如果该结点的某个子结点不存在,则用 1-1 表示。

最后一行,包含 NN 个不同的整数,表示要插入树中的数值。

输出格式

输出结果树的层序遍历序列。

数据范围

1N1001 \le N \le 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<bits/stdc++.h>

using namespace std;

const int N = 110;

int n;
int l[N], r[N];
int a[N], w[N];
int q[N];

void dfs(int u, int& k)
{
	if(u == -1)
		return;
	
	dfs(l[u], k);
	w[u] = a[k++];
	dfs(r[u], k);
}

void bfs()
{
	int hh = 0, tt = 0;
	q[0] = 0;
	
	while(hh <= tt)
	{
		int t = q[hh++];
		if(l[t] != -1)
			q[++tt] = l[t];
		if(r[t] != -1)
			q[++tt] = r[t];
	}
	
	cout << w[q[0]];
	for(int i = 1; i < n; i++)
		cout << " " << w[q[i]];
}

int main()
{	
	cin >> n;

	for(int i = 0; i < n; i++)
		cin >> l[i] >> r[i];
	for(int i = 0; i < n; i++)
		cin >> a[i];
	sort(a, a + n);	
	
	int k = 0;
	dfs(0, k);
	bfs();
	
	return 0;
}