Step1 Problem:
给你 n 个数,按照输入顺序构造二叉搜索树,然后输出每个数的父亲。
Step2 Ideas:
二叉搜索树:将要插入的数 num, 树上比 num 大的数 Max, 比 num 小的树 Min. 那么 Max 或者 Min 一定是 num 的父亲。
所以我们只用记录每个点的深度,然后求出 Max 和 Min 判断谁比较深,就可以了。
set 得用 std::set::lower_bound
std::lower_bound 对于 set 而言非常的慢
Step3 Code:
#include<bits/stdc++.h>
using namespace std;
const int N = 112345;
struct node
{
int x, id;
bool operator < (const node &b) const {
return x < b.x;
}
};
int fq[N], dep[N];
set<node> q;
set<node>::iterator lt, rt;
int main()
{
int n, num;
scanf("%d", &n);
dep[0] = 0;
for(int i = 0; i < n; i++)
{
scanf("%d", &num);
node tmp = (node){num, i};
if(i) {
rt = q.lower_bound(tmp);
lt = rt;
lt--;
if(rt == q.end()) {
fq[i] = lt->x;
dep[i] = dep[lt->id]+1;
}
else if(rt == q.begin()) {
fq[i] = rt->x;
dep[i] = dep[rt->id]+1;
}
else {
if(dep[rt->id] > dep[lt->id]) {
fq[i] = rt->x;
dep[i] = dep[rt->id]+1;
}
else {
fq[i] = lt->x;
dep[i] = dep[lt->id]+1;
}
}
}
q.insert(tmp);
}
for(int i = 1; i < n; i++)
{
printf("%d", fq[i]);
if(i == n-1) printf("\n");
else printf(" ");
}
return 0;
}