NC22596 Rinne Loves Data Structure(STL)

题目链接

题意：
$存在一棵空树$
$每次向根插入一个值，如果该值小于当前点，往左插入，否则往右$
$如果左右存在值，再次进行判断，直到空插入$
$问每次插入时所有结点的深度和$
题解：
$n<=3e5,-1e9<=val<=1e9$
$观察可得，每次插入的点，一定是和已插入点和该值相差最小的相连$
$所以用set维护树中元素$
$每次插入时进行查找他的相邻元素的深度$
$但是会发现，他可能会有两个相邻元素$
$这样的话，他肯定会去较深的元素连接$
$因为如果这两个相邻元素都在树中，他们一定有一条链连接$
$这个元素如果放在上面的点，那么一定会多出一条子树$
$但是这明显是一个二叉搜索树，不会多出子树$

$然后由于他可能没有后继或者前驱$
$我们可以找到两个哨兵放在两边$
$由于val不是1-n，所以可以考虑一下离散，这样哨兵为0和n+1$
$并且将他们的深度设为负一，就可以方便计算$
$这样一来每次相加当前深度即可$

$但是，n不是那么大，不怕多常数$
$所以干脆直接用map维护深度，就不怕val的大小没法存在数组里了$
$哨兵也可以设置为正负inf$
AC代码

/*
    Author : zzugzx
    Lang : C++
    Blog : blog.csdn.net/qq_43756519
*/
#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(), (x).end()
#define endl '\n'
#define SZ(x) (int)x.size()
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
const int mod = 1e9 + 9;
const int MOD = 998244353;
const double eps = 1e-6;
const double PI = acos(-1.0);
const int maxn = 1e6 + 10;
const int N = 1000;
const ll inf = 0x3f3f3f3f;
const int dir[][2]={{0, 1}, {1, 0}, {0, -1}, {-1, 0}, {1, 1}, {1, -1}, {-1, 1}, {-1, -1}};
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
//  freopen("in.txt", "r", stdin);
//  freopen("out.txt", "w", stdout);
    int n;
    cin >> n;
    set<int> s;
    map<int, int> d;
    s.insert(inf), s.insert(-inf);
    d[inf] = d[-inf] = -1;
    ll ans = 0;
    while (n--) {
        int x;
        cin >> x;
        auto p = s.lower_bound(x);
        d[x] = max(d[*prev(p)], d[*p]) + 1;
        s.insert(x);
        ans += d[x];
        cout << ans << endl;
    }
    return 0;
}