不懂就问(深基代码)
查看原帖
不懂就问(深基代码)
1073342
Genshin_ZFYX楼主2024/4/11 21:33
#include <iostream>
int n, cnt, root, opt, x;
struct Node {
    int right, left, value, size, num;
    Node (int l, int r, int s, int v) : right{r} , left{l}, value{v}, size{s} , num{1} {}
    Node() = default;
} t[100010];

inline void update(int root)
{
    t[root].size = t[t[root].left].size + t[t[root].right].size + t[root].num;
}

int rank(int x, int root)
{
    if (root)
    {
        if (x < t[root].value)
            return rank(x,t[root].left);
        if (x > t[root].value)
            return rank(x,t[root].right) + t[t[root].left].size + t[root].num;
        return t[t[root].left].size + t[root].num;
    }
    return 1;
}

int kth(int x, int root)
{
    if (x <= t[t[root].left].size)
        return kth(x,t[root].left);
    if (x <= t[t[root].left].size + t[root].num)
        return t[root].value;
    return kth(x - t[t[root].left].size - t[root].num, t[root].right);
}

void insert(int x, int& root)
{
    if (x < t[root].value)
        if(!t[root].left)
            t[t[root].left = ++cnt] = Node{0,0,1,x};
        else
            insert(x,t[root].left);
    else if (x > t[root].value)
        if (!t[root].right)
            t[t[root].right = ++cnt] = Node{0,0,1,x};
        else 
            insert(x,t[root].right);
    else
        t[root].num++;
    update(root);
}

int main()
{
    std::cin >> n;
    int num {};
    t[root = ++cnt] = Node {0,0,1,2147483647};
    while (n--)
    {
        std::cin >> opt >> x;
        num++;
        if (opt == 1)
            std::cout << rank(x,root) << std::endl;
        else if (opt == 2)
            std::cout << kth(x,root) << std::endl;
        else if (opt == 3)
            std::cout << kth(rank(x,root) - 1,root) << std::endl;
        else if (opt == 4)
            std::cout << kth(rank(x+1,root),root) << std::endl;
        else
        {
            num--;
            insert(x,root);
        }
    }
    
}

看了 https://www.luogu.com.cn/discuss/795437 之后还是没懂。QwQ我是小蒟蒻。

2024/4/11 21:33
加载中...