3110: [Zjoi2013]K大数查询
Time Limit: 20 Sec Memory Limit: 512 MBSubmit: 5039 Solved: 1751
[ Submit][ Status][ Discuss]
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
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
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操作中c<=Maxlongint
此题有两种做法,一种是cdq分治,一种是树套树。由于树套树简单清晰,这里就先写树套树的题解(其实是我还不会cdq分治)。
此题用线段树套线段树。外层线段树表示权值,内层为该权值范围在区间内的分布情况。这样在查询时可以用类似二分的方法得到答案,更新时则先找到外层节点,再更新其节点内层树。
在这里我用的是永久化lazy标记,主要是担心常数被卡。
#include <cstdio>
const int Maxn = 50000 + 10, MS = Maxn << 8;
int root[Maxn << 2], ls[MS], rs[MS], tot, N, M;
long long Cnt[MS], ly[MS];
//线段树的外层为权值,内层为区间
int Ina; char Inc; bool InSign;
inline int geti() {
InSign = 0;
while (Inc = getchar(), Inc < '0' || Inc > '9') InSign |= Inc == '-';
Ina = Inc - '0';
while (Inc = getchar(), Inc >= '0' && Inc <= '9') Ina = (Ina << 3) + (Ina << 1) + Inc - '0';
if (InSign) Ina = -Ina;
return Ina;
}
void Insert(int &u, int l, int r, int x, int y) {
if (!u) u = ++tot;
if (l >= x && r <= y) {
Cnt[u] += r - l + 1; ++ly[u]; return;
}
int mid = (l + r) >> 1;
if (y <= mid) Insert(ls[u], l, mid, x, y);
else if (x > mid) Insert(rs[u], mid + 1, r, x, y);
else {
Insert(ls[u], l, mid, x, mid);
Insert(rs[u], mid + 1, r, mid + 1, y);
}
Cnt[u] += y - x + 1;
}
void add(int x, int y, int val) {
int l = 1, r = N, u = 1, mid;
while (true) {
Insert(root[u], 1, N, x, y);
if (l == r) break;
mid = (l + r) >> 1; u <<= 1;
if (val <= mid) r = mid;
else l = mid + 1, u |= 1;
}
}
long long Count(int u, int l, int r, int x, int y) {
if (l >= x && r <= y) return Cnt[u];
int mid = (l + r) >> 1; long long ans = 0;
if (y <= mid) ans = Count(ls[u], l, mid, x, y);
else if (x > mid) ans = Count(rs[u], mid + 1, r, x, y);
else ans = Count(ls[u], l, mid, x, mid) + Count(rs[u], mid + 1, r, mid + 1, y);
ans += (y - x + 1) * ly[u];
return ans;
}
int query(int x, int y, long long val) {
int l = 1, r = N, mid, u = 1; long long tmp;
while (true) {
if (l == r) return l;
mid = (l + r) >> 1;
tmp = Count(root[u << 1 | 1], 1, N, x, y);
if (tmp >= val) l = mid + 1, u = u << 1 | 1;
else r = mid,u = u << 1, val -= tmp;
}
}
int main() {
N = geti(), M = geti();
int type, a, b; long long c;
while (M--) {
type = geti(), a = geti(), b = geti(), c = geti();
if (type < 2) add(a, b, c);
else printf("%d\n", query(a, b, c));
}
return 0;
}