bzoj3110【ZJOI2013】K大数查询

3110: [Zjoi2013]K大数查询

Time Limit: 20 Sec   Memory Limit: 512 MB
Submit: 3208   Solved: 1361
[ 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操作中abs(c)<=Maxlongint

Source

线段树套线段树+二分答案

树套树竟然一次就写对了，那叫一个开心啊！

外层是权值线段树，记录权值为[l,r]的信息。

内层是区间线段树，记录当前权值的位置在[l,r]的数量。

每次加点，把外层所有覆盖该权值的线段树全部更新。

每次查询，二分答案即可。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define maxn 20000005
using namespace std;
int n,m,f,a,b,c,tot;
int rt[maxn],ls[maxn],rs[maxn],sum[maxn],tag[maxn];
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
inline void update(int k,int l,int r,int x)
{
	tag[k]+=x;
	sum[k]+=(r-l+1)*x;
}
inline void pushdown(int k,int l,int r)
{
	if (!tag[k]||l==r) return;
	if (!ls[k]) ls[k]=++tot;
	if (!rs[k]) rs[k]=++tot;
	int mid=(l+r)>>1;
	update(ls[k],l,mid,tag[k]);
	update(rs[k],mid+1,r,tag[k]);
	tag[k]=0;
}
inline void pushup(int k)
{
	sum[k]=sum[ls[k]]+sum[rs[k]];
}
inline void add(int &k,int l,int r,int a,int b)
{
	if (!k) k=++tot;
	if (l==a&&r==b)
	{
		update(k,l,r,1);
		return;
	}
	pushdown(k,l,r);
	int mid=(l+r)>>1;
	if (b<=mid) add(ls[k],l,mid,a,b);
	else if (a>mid) add(rs[k],mid+1,r,a,b);
	else add(ls[k],l,mid,a,mid),add(rs[k],mid+1,r,mid+1,b);
	pushup(k);
}
inline void insert()
{
	int k=1,l=1,r=n;
	while (l!=r)
	{
		int mid=(l+r)>>1;
		add(rt[k],1,n,a,b);
		if (c<=mid) r=mid,k=k<<1;
		else l=mid+1,k=k<<1|1;
	}
	add(rt[k],1,n,a,b);
}
inline int query(int k,int l,int r,int a,int b)
{
	if (!k) return 0;
	if (l==a&&r==b) return sum[k];
	pushdown(k,l,r);
	int mid=(l+r)>>1;
	if (b<=mid) return query(ls[k],l,mid,a,b);
	else if (a>mid) return query(rs[k],mid+1,r,a,b);
	else return query(ls[k],l,mid,a,mid)+query(rs[k],mid+1,r,mid+1,b);
}
inline int solve()
{
	int k=1,l=1,r=n;
	while (l!=r)
	{
		int mid=(l+r)>>1;
		int tmp=query(rt[k<<1|1],1,n,a,b);
		if (tmp>=c) l=mid+1,k=k<<1|1;
		else c-=tmp,r=mid,k=k<<1;
	}
	return l;
}
int main()
{
	n=read();m=read();
	while (m--)
	{
		f=read();a=read();b=read();c=read();
		if (f==1) insert();
		else printf("%d\n",solve());
	}
}