HYSBZ3110-K大数查询

[Zjoi2013]K大数查询

Time Limit: 20 Sec   Memory Limit: 512 MB
Submit: 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

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操作中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;
}