bzoj 3110 K大数查询(整体二分)

题目传送门

3110: [Zjoi2013]K大数查询

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

题解

此题有两种做法，一种是整体，一种是树套树。

树套树的题解在这里。

首先我们将操作二分，将答案在l...r内的操作分在一起。如何分呢？

首先我们知道答案是在-n...n的(实际上数据只有1...n)，之后我们判断每个操作的答案与mid的关系。对于插入，如果mid的数比c大则操作在左半段，否则在右半段并跟新区间内数字个数。对于查询，如果查询区间内的数不足c，则在左半端，否则在右半段。如此递归便可获得答案。

至于维护区间有多少数，可以用线段树或树状数组(要进行区间操作)。这里我用的是树状数组。

#include <cstdio>
int Ina; char Inc; bool InSign;
inline int geti() {
	InSign = false;
	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;
}
typedef long long LL;
const int Maxn = 50000 + 5;
LL C1[2][Maxn], C2[2][Maxn], c[Maxn];
int type[Maxn], a[Maxn], b[Maxn], t1[Maxn], t2[Maxn], t[Maxn], ans[Maxn], Tim;
void Add(LL tt[2][Maxn], int x, LL val) {
	while (x < Maxn) {
		if (tt[0][x] ^ Tim) tt[0][x] = Tim, tt[1][x] = val;
		else tt[1][x] += val;
		x += x & -x;
	}
}
void Add(int x, LL val) { Add(C1, x, val); Add(C2, x, x * val); }
void Update(int l, int r, LL val) { Add(l, val); Add(r + 1, -val); }

LL Sum(LL tt[2][Maxn], int x) {
	LL ret = 0;
	while (x) {
		if (tt[0][x] >= Tim) ret += tt[1][x];
		x -= x & -x;
	}
	return ret;
}
LL Sum(const int &x) { return (x + 1)  * Sum(C1, x) - Sum(C2, x);}
LL Query(const int &l, const int &r) { return Sum(r) - Sum(l - 1); }

void Solve(int x, int y, int l, int r) {
	if (x > y) return;
	if (l == r) {
		for (int i = x; i <= y; ++i)
			if (type[t[i]] > 1) ans[t[i]] = l;
		return;
	}
	int mid = (l + r) >> 1; ++Tim; LL tmp;
	t1[0] = t2[0] = 0;
	for (int i = x; i <= y; ++i)
		if (type[t[i]] == 1) {
			if (c[t[i]] <= mid) t1[++t1[0]] = t[i];
			else t2[++t2[0]] = t[i], Update(a[t[i]], b[t[i]], 1);
		} else {
			if ((tmp = Query(a[t[i]], b[t[i]])) < c[t[i]]) c[t[i]] -= tmp, t1[++t1[0]] = t[i];
			else t2[++t2[0]] = t[i];
		}
	int midd = x + t1[0] - 1;
	for (int i = x; i <= midd; ++i) t[i] = t1[i - x + 1];
	for (int i = midd + 1; i <= y; ++i) t[i] = t2[i - midd];
	Solve(x, midd, l, mid); Solve(midd+1, y, mid + 1, r);
}

int main() {
	int n = geti(), m= geti();
	for (int i = 1; i <= m; ++i) {
		type[i] = geti(), a[i] = geti(), b[i] = geti(), c[i] = geti(); t[i] = i;
	}
	Solve(1, m, 1, n);
	for (int i = 1; i <= m; ++i)
		if (type[i] > 1) printf("%d\n", ans[i]);
	return 0;
}

附上关于树状数组的区间操作的博客： http://blog.csdn.net/lawrence_jang/article/details/8054173