洛谷P3960 列队(动态开节点线段树)

题意

题目链接

sol

看不懂splay。。，看不懂树状数组。。。

只会暴力动态开节点线段树

观察之后不难发现，我们对于行和列需要支持的操作都是相同的：找到第\(k\)大的元素并删除，在末尾插入一个元素

这样我们可以维护\(n+1\)棵线段树(对列单独建一棵)

每次操作的时候，如果\(y_i = m\)，那么只对列所在的线段树进行操作

否则，首先在第\(x_i\)棵线段树中找到第\(y_i\)大的元素并删除，在列所在的线段树中找到需要插入的元素并记录下来。

然后再删除列中对应的元素

当然，还有许多很巧妙的操作。

1. 对于没有被更改过的点，我们可以\(o(1)\)计算出它的编号。显然，被更改过的点不会太多，直接拿vector维护即可

2. 开始时每个线段树上的对应节点的siz都是满的，所以对于没有更新过的区间，直接拿区间长度来获取siz即可

3. 我们在查找到某个节点的时候同时也会删除它，因此查找和删除操作可以写到同一个函数中

#include<bits/stdc++.h>
#define ll long long int
#define getchar() (p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 1 << 21, stdin), p1 == p2) ? eof : *p1++)
char buf[1 << 23], *p1 = buf, *p2 = buf,  obuf[1 << 24], *o = obuf;
void print(ll x) {
    if(x > 9) print(x / 10);
    *o++= x % 10 + '0';
}
using namespace std;
const int maxn = 3e5 + 10, ss = maxn * 20;
inline int read() {
    char c = getchar(); int x = 0, f = 1;
    while(c < '0'|| c > '9') {if(c == '-') f = -1; c = getchar();}
    while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return x * f;
}
int n, m, q, cnt, siz[ss], ls[ss], rs[ss], rt[maxn], tot;
ll las[maxn << 1];
vector<ll> pos[maxn];
int modify(int &k, int l, int r, int kth) {
    if(!k) k = ++tot, siz[k] = r - l + 1;  siz[k]--;
    if(l == r) return l;
    int mid = l + r >> 1, now = (!ls[k] ? mid - l + 1 : siz[ls[k]]);
    if(now < kth) return modify(rs[k], mid + 1, r, kth - now);
    else return modify(ls[k], l, mid, kth);
}
ll query(int y, int x) {
    ll ans = 0;
    if(y == m) ans = las[modify(rt[0], 1, n + q, x)];
    else {
        int tmp = modify(rt[x], 1, m + q, y);
        if(tmp < m) ans = 1ll * (x - 1) * m + tmp;
        else ans = pos[x][tmp - m];
        pos[x].push_back(las[modify(rt[0], 1, n + q, x)]);
    }
    return las[++cnt] = ans;
}
int main() {
    n = read(); m = read(); q = read();
    for(int i = 1; i <= n; i++) las[++cnt] = 1ll * i * m;
    for(int i = 1; i <= q; i++) print(query(read(), read())), *o++ = '\n';
    fwrite(obuf, o-obuf, 1 , stdout);
    return 0;
}