[Zjoi2013]K大数查询
Time Limit: 20 Sec Memory Limit: 512 MBSubmit: 9350 Solved: 2761
[ 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
Source
解题思路:线段树套线段树,外面一棵是权值线段树,里面一棵是区间线段树
#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <cctype>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <vector>
#include <bitset>
#include <functional>
using namespace std;
#define LL long long
const int INF = 0x3f3f3f3f;
const int maxn = 50009;
struct node
{
int op, l, r;
long long z;
}a[maxn];
long long x[maxn];
int n, m, cnt, tot;
int s[200005], L[20000005], R[20000005];
long long sum[20000005], lazy[20000005];
void Push(int k, int l, int r)
{
if (!L[k]) L[k] = ++tot;
if (!R[k]) R[k] = ++tot;
lazy[L[k]] += lazy[k]; lazy[R[k]] += lazy[k];
int mid = (l + r) >> 1;
sum[L[k]] += 1LL * (mid - l + 1)*lazy[k];
sum[R[k]] += 1LL * (r - mid)*lazy[k];
lazy[k] = 0;
}
void update(int &k, int l, int r, int ll, int rr)
{
if (!k) k = ++tot;
if (ll <= l&&rr >= r)
{
sum[k] += 1LL * (r - l + 1);
lazy[k]++;
return;
}
if(lazy[k]) Push(k, l, r);
int mid = (l + r) >> 1;
if (ll <= mid) update(L[k], l, mid, ll, rr);
if (rr > mid) update(R[k], mid + 1, r, ll, rr);
sum[k] = sum[L[k]] + sum[R[k]];
}
void Insert(int ll, int rr, int p)
{
int k = 1, l = 1, r = cnt - 1;
while (l != r)
{
int mid = (l + r) >> 1;
update(s[k], 1, n, ll, rr);
if (p <= mid) r = mid, k = k << 1;
else l = mid + 1, k = k << 1 | 1;
}
update(s[k], 1, n, ll, rr);
}
LL query(int k, int l, int r, int ll, int rr)
{
if (!k) return 0;
if (ll <= l&&rr >= r) return sum[k];
int mid = (l + r) >> 1;
LL ans = 0;
if(lazy[k]) Push(k, l, r);
if (ll <= mid) ans += query(L[k], l, mid, ll, rr);
if (rr > mid) ans += query(R[k], mid + 1, r, ll, rr);
return ans;
}
int solve(int ll, int rr, LL p)
{
int l = 1, r = cnt - 1, k = 1;
while (l != r)
{
int mid = (l + r) >> 1;
LL temp = query(s[k << 1 | 1], 1, n, ll, rr);
if (temp >= p) l = mid + 1, k = k << 1 | 1;
else r = mid, k <<= 1, p -= temp;
}
return l;
}
int main()
{
scanf("%d%d", &n, &m);
for (int i = 1; i <= m; i++) scanf("%d%d%d%lld", &a[i].op, &a[i].l, &a[i].r, &a[i].z), x[i] = a[i].z;
sort(x + 1, x + 1 + m);
cnt = unique(x + 1, x + 1 + m) - x;
for (int i = 1; i <= m; i++)
{
if (a[i].op == 1)
{
int k = lower_bound(x + 1, x + cnt, a[i].z) - x;
Insert(a[i].l, a[i].r, k);
}
else printf("%lld\n", x[solve(a[i].l, a[i].r, a[i].z)]);
}
return 0;
}