BZOJ 3110 ZJOI 2013 K大值查询 线段树套线段树

题目：

有N个位置，M个操作。操作有两种，每次操作如果是1 a b c的形式表示在第a个位置到第b个位置，每个位置加入一个数c
如果是2 a b c形式，表示询问从第a个位置到第b个位置，第C大的数是多少。

先维护权值再维护区间？

反正数字最大就5W?

对权值建立[1,n]的权值线段树，而且是单点修改单点查询。

每个节点维护拥有该权值的区间线段树。

[a,b]加数字c的话，在权值线段树中包含c的节点对应的[a,b]区间全部加一遍就好。

查询的话就像二分查找就好了。

太神奇了，如果把interval_add和interval_query的ql和qr放到函数的参数里会T的好惨。。

原来430ms一个点要跑10s？！

我记得应该没有这么大的性能消耗的。。

难道我记错了。。

看着Rank 1的500ms 望尘莫及。。

#include <cstdio>
#include <algorithm>
using namespace std;
const int N = 50001, M = 5000000;
int lc[M], rc[M], sum[M], t[M], tree[N * 3], cnt;
int n, ql, qr;
void interval_add(int &x, int l, int r) {
    if (!x) x = ++cnt;
    if (ql <= l && r <= qr) { sum[x] += r - l + 1; t[x] ++; return;}
    int m = l + r >> 1;
    if (ql <= m) interval_add(lc[x], l, m);
    if (qr > m) interval_add(rc[x], m + 1, r);
    sum[x] += min(qr, r) - max(ql, l) + 1;
}
int interval_query(int x, int l, int r) {
    if (!x) return 0;
    if (ql <= l && r <= qr) return sum[x];
    int m = l + r >> 1, ret = 0;
    if (ql <= m) ret += interval_query(lc[x], l, m);
    if (qr > m) ret += interval_query(rc[x], m + 1, r);
    return ret + t[x] * (min(qr, r) - max(ql, l) + 1);
}
void weight_add(int a, int b, int c) {
    int l = 1, r = n, id = 1;
    ql = a; qr = b;
    while (interval_add(tree[id], 1, n), l < r) {
        int m = l + r >> 1;
        if (c <= m) r = m, id = id * 2;
        else l = m + 1, id = id * 2 + 1;
    }
}
int weight_kth(int a, int b, int c) {
    int l = 1, r = n, id = 1;
    ql = a; qr = b;
    while (l < r) {
        int m = l + r >> 1;
        int tmp = interval_query(tree[id * 2 + 1], 1, n);
        if (c <= tmp) l = m + 1, id = id * 2 + 1;
        else r = m, c -= tmp, id = id * 2;
    }
    return l;
}
int main() {
    int op, a, b, c, i, m;
    scanf("%d%d", &n, &m);
    for (i = 1; i <= m; i++) {
        scanf("%d%d%d%d", &op, &a, &b, &c);
        if (op == 1) weight_add(a, b, c);
        else printf("%d\n", weight_kth(a, b, c));
    }
    return 0;
}

3110: [Zjoi2013]K大数查询

Time Limit: 20 Sec   Memory Limit: 512 MB
Submit: 2826   Solved: 1211

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