bzoj 3110 K大数查询(树套树)

题目传送门

3110: [Zjoi2013]K大数查询

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

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

题解

此题有两种做法，一种是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;
}