[BZOJ]3110: [Zjoi2013]K大数查询 整体二分+线段树

Description

有N个位置，M个操作。操作有两种，每次操作如果是1 a b c的形式表示在第a个位置到第b个位置，每个位置加入一个数c

如果是2 a b c形式，表示询问从第a个位置到第b个位置，第C大的数是多少。

题解：

这题跟入门题POJ2104差不多，不过那道题目是改点求段，这道题目是改段求段，用线段树代替树状数组就可以了。

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=50010;
const int inf=50000;
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<<3)+(x<<1)+(ch^48),ch=getchar();
return x*f;
}
int n,m,cnt=0,ans[Maxn],trlen=0;
struct Opt{int type,x,y,z,id;}q[Maxn],q1[Maxn],q2[Maxn];
struct Seg{int l,r,lc,rc;LL s,lazy;}tr[Maxn<<1];
void build(int l,int r)
{
int t=++trlen;
tr[t].l=l;tr[t].r=r
4000
;tr[t].s=tr[t].lazy=0;
if(l<r)
{
int mid=l+r>>1;
tr[t].lc=trlen+1;build(l,mid);
tr[t].rc=trlen+1;build(mid+1,r);
}
}
void work(Seg &x,LL y){x.lazy+=y;x.s+=(LL)((x.r-x.l+1)*y);}
void push_down(Seg &x)
{
LL t=x.lazy;x.lazy=0;
work(tr[x.lc],t);work(tr[x.rc],t);
}
void add(int now,int l,int r,LL x)
{
if(tr[now].l==l&&tr[now].r==r){work(tr[now],x);return;}
int lc=tr[now].lc,rc=tr[now].rc,mid=tr[now].l+tr[now].r>>1;
if(tr[now].lazy!=0)push_down(tr[now]);
if(r<=mid)add(lc,l,r,x);
else if(l>mid)add(rc,l,r,x);
else add(lc,l,mid,x),add(rc,mid+1,r,x);
tr[now].s=tr[lc].s+tr[rc].s;
}
LL query(int now,int l,int r)
{
if(tr[now].l==l&&tr[now].r==r)return tr[now].s;
int lc=tr[now].lc,rc=tr[now].rc,mid=tr[now].l+tr[now].r>>1;
if(tr[now].lazy!=0)push_down(tr[now]);
if(r<=mid)return query(lc,l,r);
else if(l>mid)return query(rc,l,r);
else return query(lc,l,mid)+query(rc,mid+1,r);
tr[now].s=tr[lc].s+tr[rc].s;
}
void solve(int l,int r,int L,int R)
{
if(l>r)return;
if(L==R)
{
for(int i=l;i<=r;i++)
if(q[i].type==2)ans[q[i].id]=L;
return;
}
int Mid=L+R>>1,l1=0,l2=0;
for(int i=l;i<=r;i++)
{
if(q[i].type==1)
{
if(q[i].z>Mid)add(1,q[i].x,q[i].y,1),q2[++l2]=q[i];
else q1[++l1]=q[i];
}
else
{
LL t=query(1,q[i].x,q[i].y);
if(t<q[i].z)q[i].z-=t,q1[++l1]=q[i];
else q2[++l2]=q[i];
}
}
for(int i=1;i<=l1;i++)q[l+i-1]=q1[i];
for(int i=1;i<=l2;i++)q[l+i+l1-1]=q2[i];
for(int i=1;i<=l2;i++)if(q2[i].type==1)add(1,q2[i].x,q2[i].y,-1);
solve(l,l+l1-1,L,Mid);solve(l+l1,r,Mid+1,R);
}
int main()
{
n=read(),m=read();
for(int i=1;i<=m;i++)
{
q[i].type=read(),q[i].x=read(),q[i].y=read(),q[i].z=read();
if(q[i].type==2)q[i].id=++cnt;
}
build(1,n);
solve(1,m,-inf,inf);
for(int i=1;i<=cnt;i++)printf("%d\n",ans[i]);
}