区间求第k小,变为了区间求第k大。
那么现在,对于插入数的结构体来说,如果q[i].k>mid,就把它划分到右区间,并进行单点累加;如果q[i].k<=mid,则划分到左区间。而对于查询的结构体来说,如果查询区间内的值,记为sum,当q[i].k>sum时,可知是由于mid不够小的缘故,所以应该把mid再变小写,即划分到左区间;当q[i].k<=sum时,说明mid已经够大了,所有划分到右区间。
这里的划分,和求第k小相比,好像不是那么“显然”与“对称”了,所以需要注意。
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N=5e4+5;
int n,m,Q;
int sum[N<<2],tag[N<<2],ans[N];
struct node{int x,y,k,id,opt;}q[N<<1],q1[N<<1],q2[N<<1];
inline void pushdown(int k,int l,int r,int mid)
{
if (tag[k])
{
tag[k<<1]+=tag[k]; tag[k<<1|1]+=tag[k];
sum[k<<1]+=(mid-l+1)*tag[k];
sum[k<<1|1]+=(r-(mid+1)+1)*tag[k];
tag[k]=0;
}
}
void change(int k,int l,int r,int qx,int qy,int v)
{
if (qx<=l && r<=qy)
{
sum[k]+=(r-l+1)*v;
tag[k]+=v;
return;
}
int mid=l+r>>1;
pushdown(k,l,r,mid);
if (qx<=mid) change(k<<1,l,mid,qx,qy,v);
if (mid<qy) change(k<<1|1,mid+1,r,qx,qy,v);
sum[k]=sum[k<<1]+sum[k<<1|1];
}
int query(int k,int l,int r,int qx,int qy)
{
if (qx<=l && r<=qy) return sum[k];
int mid=l+r>>1;
pushdown(k,l,r,mid);
int res=0;
if (qx<=mid) res+=query(k<<1,l,mid,qx,qy);
if (mid<qy) res+=query(k<<1|1,mid+1,r,qx,qy);
return res;
}
void solve(int ql,int qr,int l,int r)
{
if (ql>qr) return;
if (l==r)
{
for (register int i=ql; i<=qr; ++i) if (q[i].opt==2) ans[q[i].id]=l;
return;
}
int mid=l+r>>1;
int p1=0,p2=0;
for (register int i=ql; i<=qr; ++i)
if (q[i].opt==1)
{
if (q[i].k>mid)
{
change(1,1,n,q[i].x,q[i].y,1);
q2[++p2]=q[i];
}
else q1[++p1]=q[i];
}
else
{
int res=query(1,1,n,q[i].x,q[i].y);
if (q[i].k>res)
{
q[i].k-=res;
q1[++p1]=q[i];
}
else
{
q2[++p2]=q[i];
}
}
for (register int i=1; i<=p2; ++i) if (q2[i].opt==1) change(1,1,n,q2[i].x,q2[i].y,-1);
for (register int i=1; i<=p1; ++i) q[ql+i-1]=q1[i];
for (register int i=1; i<=p2; ++i) q[ql+p1-1+i]=q2[i];
solve(ql,ql+p1-1,l,mid);
solve(ql+p1,qr,mid+1,r);
}
signed main(){
scanf("%lld%lld",&n,&m);
for (register int i=1; i<=m; ++i)
{
scanf("%lld%lld%lld%lld",&q[i].opt,&q[i].x,&q[i].y,&q[i].k);
if (q[i].opt==2) q[i].id=++Q;
}
solve(1,m,-n,n);
for (register int i=1; i<=Q; ++i) printf("%d\n",ans[i]);
return 0;
}