PAT--1099 Build A Binary Search Tree--中序遍历

题目链接:https://pintia.cn/problem-sets/994805342720868352/problems/994805367987355648

题目大意:给你一棵二叉搜索树,然后再给你这个树的数组,让你去还原它,并且按层序遍历输出。

分析:二叉搜索树中序遍历,得到的就是排好序的数组,所以我们把数组排个序,按中序遍历放到树中就行了(都快忘了中序遍历这个东西了,导致想了好久。。。自闭)。

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
const int N = 105;
int n, cnt, a[N], l[N], r[N], ans[N];
void dfs(int x) {
	if(l[x] != -1) dfs(l[x]);
	ans[x] = a[cnt++];
	if(r[x] != -1) dfs(r[x]);
}
void print() {
	queue<int> q;
	q.push(0);
	int d = 0;
	while(!q.empty()) {
		int x = q.front();
		q.pop();
		if(d == 0) printf("%d", ans[x]);
		else printf(" %d", ans[x]);
		if(l[x] != -1) q.push(l[x]);
		if(r[x] != -1) q.push(r[x]);
		d++;
	}
	printf("\n");
}
int main() {
	scanf("%d", &n);
	memset(l, -1, sizeof l);
	memset(r, -1, sizeof r);
	for(int i = 0; i < n; i++)
		scanf("%d %d", &l[i], &r[i]);
	for(int i = 1; i <= n; i++)
		scanf("%d", &a[i]);
	sort(a + 1, a + n + 1);
	cnt = 1;
	dfs(0);
	print();
	return 0;
}
发布了150 篇原创文章 · 获赞 4 · 访问量 6960

猜你喜欢

转载自blog.csdn.net/Napom/article/details/102745005