P3850 [TJOI2007]书架 （splay维护序列）

传送门

1.题意

Knuth 先生家里有个精致的书架，书架上有 NN 本书，如今他想学到更多的知识，于是又买来了 MM 本不同的新书。现在他要把新买的书依次插入到书架中，他已经把每本书要插入的位置标记好了，并且相应的将它们放好。由于 Knuth 年龄已大，过几天他已经记不清某些位置上放的到底是什么书了，请问你能帮助他吗？

2.分析

说白了，就是不停的向序列中插入元素（插队）。最后查询第i个位置是什么。
splay可以很简单的进行插入操作。

3.代码

#include <bits/stdc++.h>

using namespace std;
//-----pre_def----
const double PI = acos(-1.0);
const int INF = 0x3f3f3f3f;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
#define fir(i, a, b) for (int i = (a); i <= (b); i++)
#define rif(i, a, b) for (int i = (a); i >= (b); i--)
#define endl '\n'
#define init_h memset(h, -1, sizeof h), idx = 0;
#define lowbit(x) x &(-x)

//---------------
const int N = 1e5 + 210;
int n, m, q, root, idx, x;
string d;
struct node
{
    int s[2], p, size;
    string v;
} tr[N];
void pushup(int u)
{
    tr[u].size = tr[tr[u].s[0]].size + tr[tr[u].s[1]].size + 1;
}
void rotate(int x) //核心函数1：将所有情况的左旋右旋写在一个函数里面
{
    int y = tr[x].p, z = tr[y].p; //在这个链上：z是祖宗，y是爸爸，x是儿子。现在要把x移动到最上边
    int k = tr[y].s[1] == x;      // k=0表示x是y的左儿子；k=1表示x是y的右儿子
    tr[z].s[tr[z].s[1] == y] = x;
    tr[x].p = z;
    tr[y].s[k] = tr[x].s[k ^ 1];
    tr[tr[x].s[k ^ 1]].p = y;
    tr[x].s[k ^ 1] = y;
    tr[y].p = x;
    pushup(y), pushup(x); //顺序不能乱
}
void splay(int x, int k) //核心函数2：将x节点旋转至k下面,如果k是0则旋转到根
{
    while (tr[x].p != k) //但x的父节点不为k时
    {
        int y = tr[x].p, z = tr[y].p;
        if (z != k) //x的祖宗还不是k
        {
            if ((tr[y].s[1] == x) ^ (tr[z].s[1] == y)) //异或，当zyx不为一条直链的时候
            {
                rotate(x);
            }
            else
            {
                rotate(y);
            }
        }
        rotate(x);
    }
    if (!k) //如果k==0，等加于将x旋转至root
        root = x;
}
void insert(string x)
{
    tr[root].s[1] = ++idx;
    tr[idx].v = x;
    tr[idx].p = root;
    tr[idx].size = 1;
    splay(idx, 0);
}
int find_k(int k)
{
    int u = root;
    while (u)
    {
        if (tr[tr[u].s[0]].size >= k)
            u = tr[u].s[0];
        else if (tr[tr[u].s[0]].size + 1 == k)
            return u;
        else
            k -= tr[tr[u].s[0]].size + 1, u = tr[u].s[1];
    }
    return -1;
}
void insert(string d, int x) //将d插入到x位置
{
    int l = find_k(x - 1);
    int r = find_k(x);
    splay(l, 0);
    splay(r, l);
    tr[++idx].v = d;
    tr[idx].p = r;
    tr[r].s[0] = idx;
    tr[idx].size = 1;
    pushup(r), pushup(l);
}
void print(int u = root)
{
    if (tr[u].s[0])
        print(tr[u].s[0]);
    cout << tr[u].v << " ";
    if (tr[u].s[1])
        print(tr[u].s[1]);
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
    int StartTime = clock();
#endif
    scanf("%d", &n);
    insert("begin");//哨兵
    fir(i, 1, n)
    {
        cin >> d;
        insert(d);
    }
    insert("end");//哨兵
    scanf("%d", &m);
    while (m--)
    {
        cin >> d >> x;
        insert(d, x + 2);
    }
    // print();
    // puts("");
    scanf("%d", &q);
    while (q--)
    {
        scanf("%d", &x);
        splay(find_k(x + 2), 0);
        cout << tr[root].v << endl;
    }

#ifndef ONLINE_JUDGE
    printf("Run_Time = %d ms\n", clock() - StartTime);
#endif
    return 0;
}