[BZOJ3110][Zjoi2013]K大数查询

[Zjoi2013]K大数查询

Description
有N个位置，M个操作。操作有两种，每次操作如果是1 a b c的形式表示在第a个位置到第b个位置，每个位置加入一个数c
如果是2 a b c形式，表示询问从第a个位置到第b个位置，第C大的数是多少。
Input
第一行N，M
接下来M行，每行形如1 a b c或2 a b c
Output
输出每个询问的结果
Sample Input
2 5
1 1 2 1
1 1 2 2
2 1 1 2
2 1 1 1
2 1 2 3
Sample Output
1
2
1
HINT
【样例说明】
第一个操作 后位置 1 的数只有 1 ， 位置 2 的数也只有 1 。 第二个操作 后位置 1
的数有 1 、 2 ，位置 2 的数也有 1 、 2 。 第三次询问 位置 1 到位置 1 第 2 大的数 是
1 。 第四次询问 位置 1 到位置 1 第 1 大的数是 2 。 第五次询问 位置 1 到位置 2 第 3
大的数是 1 。‍
N,M<=50000,N,M<=50000
a<=b<=N
1操作中abs(c)<=N
2操作中abs(c)<=Maxlongint

Solution
本来想树状数组套权值线段树的，发现空间炸裂，无奈只能权值套区间线段树。
外层是权值线段树，维护的是权值在[l,r]之间的数出现在了哪些区间，这个哪些区间就用挂在它下面的区间线段树来维护。
然后我们二分就可以做了。

Code

#include <bits/stdc++.h>using namespace std;#define rep(i, l, r) for (int i = (l); i <= (r); i++)#define per(i, r, l) for (int i = (r); i >= (l); i--)#define X first#define Y second#define PB push_back#define MP make_pair#define MS(_) memset(_, 0, sizeof(_))#define debug(...) fprintf(stderr, __VA_ARGS__)typedef unsigned int ll;const int N = 50000 + 10;struct node{    ll sum, tag;    node *lc, *rc;    void modify(int l, int r, ll d) {sum += 1ll * (r-l+1) * d; tag += d;}    void sink(int l, int r);    void update() {sum = (lc ? lc->sum : 0) + (rc ? rc->sum : 0);}}pool[N * 600], *tail = pool;inline node *newnode() {return tail++;}inline void node::sink(int l, int r){    if (!lc) lc = newnode(); lc->modify(l, l+r>>1, tag);    if (!rc) rc = newnode(); rc->modify((l+r>>1)+1, r, tag);    tag = 0;}struct NODE{     node *root;    NODE *LC, *RC;}POOL[N << 2],    *ROOT, *TAIL = POOL;inline NODE *NEWNODE() {return TAIL++;}int n, m;template<typename T> inline void read(T &x){    x = 0; T f = 1; char ch = getchar();    while (!isdigit(ch)) {if (ch == '-') f = -1; ch = getchar();}    while (isdigit(ch))  {x = x * 10 + ch - '0'; ch = getchar();}    x *= f;}inline void BUILD(NODE *(&RT), int L, int R){    RT = NEWNODE();  RT->root = newnode();    if (L == R) return;    int MID = L+R >> 1;    BUILD(RT->LC, L, MID); BUILD(RT->RC, MID+1, R);}inline void insert(node *rt, int l, int r, int ql, int qr){    rt->sink(l, r);    if (ql <= l && r <= qr) rt->modify(l, r, 1);    else{        int mid = l+r >> 1;        if (ql <= mid){            if (!rt->lc) rt->lc = newnode(); insert(rt->lc, l, mid, ql, qr);        }        if (qr > mid){            if (!rt->rc) rt->rc = newnode(); insert(rt->rc, mid+1, r, ql, qr);        }        rt->update();    }}inline void INSERT(NODE *RT, int L, int R, int ql, int qr, int pos){    insert(RT->root, 1, n, ql, qr);    if (L == R) return;    int MID = L+R >> 1;    if (pos <= MID) INSERT(RT->LC, L, MID, ql, qr, pos);    else INSERT(RT->RC, MID+1, R, ql, qr, pos);}inline ll query(node *rt, int l, int r, int ql, int qr){    rt->sink(l, r);    if (ql <= l && r <= qr) return rt->sum;    int mid = l+r >> 1; ll res = 0;    if (ql <= mid && rt->lc) res += query(rt->lc, l, mid, ql, qr);    if (qr > mid && rt->rc) res += query(rt->rc, mid+1, r, ql, qr);    return res;}inline void QUERY(NODE *RT, int L, int R, int ql, int qr, int rank){    if (L == R) {printf("%d\n", L); return;}    ll cnt = query(RT->RC->root, 1, n, ql, qr);    //debug("X = %d, ql = %d, qr = %d, cnt = %d\n", (L+R) >> 1, ql, qr, cnt);    int MID = L+R >> 1;    if (cnt < rank) QUERY(RT->LC, L, MID, ql, qr, rank-cnt);    else QUERY(RT->RC, MID+1, R, ql, qr, rank);}int main(){    read(n); read(m);    BUILD(ROOT, 1, n);    rep(i, 1, m){ int op, a, b, c;        read(op); read(a); read(b); read(c);        if (op == 1) INSERT(ROOT, 1, n, a, b, c);        else QUERY(ROOT, 1, n, a, b, c);    }    return 0;}