【nlogn 构造二叉搜索树】CodeForces - 675D Tree Construction

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;
}