题目链接: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;
}