AcWing 5061. 二叉搜索树 二叉搜索树 map

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 100010;

int root=0;
int l[N],r[N];
int fa[N];
int n;

int insert(int pre,int node,int x){
    if(!node){
        fa[x] = pre;
        return x;
    }
    if(node > x){
        l[node] = insert(node,l[node],x);
    }else{
        r[node] = insert(node,r[node],x);
    }
    return node;
}


int main()
{
    int m;
    cin>> n;
    for (int i = 1; i <= n; i ++ ){
        cin>>m;
        root = insert(0,root,m);
    }
    for (int i = 1; i <= n; i ++ ){
        cout << fa[i]<<" ";
    }

    return 0;
}

#include <bits/stdc++.h>
using namespace std;

const int N = 100010;
int n;
map<int ,int> d;//存这个点和这个点所在的层数（深度）
int p[N];//存一下每个点的根节点是谁

signed main()
{
    cin>>n;
    //初始化两个哨兵，因为一开始什么都没有 a(?) x b(?)
    d[0] = 0,d[n+1] = -1;//这样插入第一个点的时候就有了参考
    //因为是1~n 所以选择的哨兵是0 和 n+1
    //第一个哨兵初始化为0（当树为空的时候，优先选择0号点为父节点）
    //那个初始为-1 因为要选深度大的为父节点所以初始化比0小就行
    //这样设置就满足题意，第一个点进来的父节点为0
    for(int i = 0 ; i < n ; i++)
    {
        int x;
        cin>>x;
        //求一下比x大的最小的数
        auto b = d.upper_bound(x), a = b;
        //比x小的最大的数（b的前一个数，就是比x小的最大的数)
        --a;

        //判断一下谁的深度大
        //如果a的深度大
        if(a->second > b-> second)
        {
            d[x] = a-> second + 1;//(把x插到a下面)
            p[x] = a-> first;//x的父节点是a
        }
        //b的深度大
        else
        {
            d[x] = b -> second + 1;
            p[x] = b -> first;
        }
    }
    for(int i = 1 ; i <= n ; i++)
    {
        cout<<p[i]<<" ";
    }
    return 0;
}