1. 程式人生 > >洛谷P3960 列隊(動態開節點線段樹)

洛谷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;
}