[bzoj3110][Zjoi2013]K大数查询

来自FallDream的博客，未经允许，请勿转载，谢谢。

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

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

这道题数据结构随便切吧 什么树套树之类的

学习一下整体二分，写了个模板，查询线段树/树状数组。整体二分啥具体的可以自行百度，不难理解吧..

但是感觉我写的好丑啊常数巨大

#include<iostream>
#include<cstdio>
#define MN 50000
#define INF 2147483647
#define ll long long
using namespace std;
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;
}

int n,m;ll ans[MN+5];
struct op{int kind,a,b;ll c;int id;}q[MN*60];
struct Tree{int l,r;ll val,x;}T[MN*14+5];

void pushdown(int x)
{
int l=x<<1,r=x<<1|1;
T[l].x+=1LL*(T[l].r-T[l].l+1)*T[x].val;
T[r].x+=1LL*(T[r].r-T[r].l+1)*T[x].val;
T[l].val+=T[x].val;T[r].val+=T[x].val;
T[x].val=0;
}

void build(int x,int l,int r)
{
if((T[x].l=l)==(T[x].r=r)) return;
int mid=l+r>>1;
build(x<<1,l,mid);
build(x<<1|1,mid+1,r);
}

void renew(int x,int l,int r,int ad)
{
if(T[x].l==l&&T[x].r==r)
{
T[x].x+=1LL*(r-l+1)*ad;
T[x].val+=ad;return;
}
if(T[x].val) pushdown(x);
int mid=T[x].l+T[x].r>>1;
if(r<=mid) renew(x<<1,l,r,ad);
else if(l>mid) renew(x<<1|1,l,r,ad);
else renew(x<<1,l,mid,ad),renew(x<<1|1,mid+1,r,ad);
T[x].x=T[x<<1].x+T[x<<1|1].x;
}

ll query(int x,int l,int r)
{
if(T[x].l==l&&T[x].r==r) return T[x].x;
if(T[x].val) pushdown(x);
int mid=T[x].l+T[x].r>>1;
if(r<=mid) return query(x<<1,l,r);
else if(l>mid) return query(x<<1|1,l,r);
else return query(x<<1,l,mid)+query(x<<1|1,mid+1,r);
}

void solve(int lt,int rt,int l,int r)
{
if(l==r)
{
for(int i=lt;i<=rt;i++)
if(q[i].kind>1) ans[q[i].id]=l;
return;
}
int mid=((ll)l+(ll)r+1)>>1,beg=m+1;
for(int i=lt;i<=rt;i++) if(q[i].kind>1) ans[q[i].id]=q[i].c+query(1,q[i].a,q[i].b);
for(int i=lt;i<=rt;i++)
{
if(q[i].kind==1) q[i].c>=mid?(renew(1,q[i].a,q[i].b,1),q[++m]=q[i],0):0;
else
{
ans[q[i].id]-=query(1,q[i].a,q[i].b);
if(ans[q[i].id]<=0) q[++m]=q[i];
else q[i].c=ans[q[i].id];
}
}
int ed=m;
for(int i=lt;i<=rt;i++)
if((ans[q[i].id]>0&&q[i].kind>1)||(q[i].kind==1&&q[i].c<mid))
q[++m]=q[i];
int end=m;
if(beg<=ed)solve(beg,ed,mid,r);
if(end>ed) solve(ed+1,end,l,mid-1);
}

int main()
{
n=read();m=read();build(1,1,n);
for(int i=1;i<=m;i++)
{
q[i].kind=read();q[i].a=read();q[i].b=read();
q[i].c=read();q[i].id=i;
}
int tot=m;
solve(1,m,-INF,INF);
for(int i=1;i<=tot;i++)if(q[i].kind==2)printf("%lld\n",ans[i]);
return 0;
}