bzoj3110【ZJOI2013】K大数查询

Description

Input

Output

Sample Input

Sample Output

HINT

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,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 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 a,int b)
{
	if (!k) k=++tot;
	if (l==a&&r==b)
	{
		update(k,1);
		return;
	}
	pushdown(k,r);
	int mid=(l+r)>>1;
	if (b<=mid) add(ls[k],b);
	else if (a>mid) add(rs[k],b);
	else add(ls[k],mid),add(rs[k],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,b);
		if (c<=mid) r=mid,k=k<<1;
		else l=mid+1,k=k<<1|1;
	}
	add(rt[k],b);
}
inline int query(int k,int b)
{
	if (!k) return 0;
	if (l==a&&r==b) return sum[k];
	pushdown(k,r);
	int mid=(l+r)>>1;
	if (b<=mid) return query(ls[k],b);
	else if (a>mid) return query(rs[k],b);
	else return query(ls[k],mid)+query(rs[k],b);
}
inline int solve()
{
	int k=1,r=n;
	while (l!=r)
	{
		int mid=(l+r)>>1;
		int tmp=query(rt[k<<1|1],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("%dn",solve());
	}
}