3110: [Zjoi2013]K大数查询
Time Limit: 20 Sec Memory Limit: 512 MBSubmit: 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
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
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());
}
}