题面:传送门
一看就是一道毒瘤数据结构题,因为既有修改又有第 k k 大查询,所以我们可以考虑树套树,但是线段树套平衡树是显然不可行的,每次修改的时候懒标记可能会使空间复杂度变为。考虑线段树套线段树,那这题一定是一棵区间线段树和一棵权值线段树,那谁在里谁在外呢?由于每次都是将一个区间全部加入一个相同的数,所以权值线段树必须在外。
这道题因为要动态开点还是比较难写的,我 wa w a 了好几发,调了半天才调对。由于是两棵线段树,所以修改的复杂度是 Θ(mlog22n) Θ ( m l o g 2 2 n ) ,查询二分答案的过程可以与权值线段树合并,所以复杂度也是 Θ(mlog22n) Θ ( m l o g 2 2 n ) 。最终复杂度: Θ(mlog22n) Θ ( m l o g 2 2 n ) 。(据说整体二分更快,但我不会QAQ)
全部代码:
#pragma GCC optimize(2)
#include<iostream>
#include<cstdio>
#include<cstdlib>
using namespace std;
int n,m;
namespace sgt2
{
long long sum[20000010];
int chg[20000010],son[20000010][2],cnt;
int nw(int &x)
{
if(!x)return x=++cnt;
return x;
}
void update(int a,int b,int k,int l,int r)
{
if(a==l && b==r)
{
chg[k]++;
sum[k]+=r+1-l;
}
else
{
int mid=(l+r)>>1;
if(b<=mid)update(a,b,nw(son[k][0]),l,mid);
else if(a>mid)update(a,b,nw(son[k][1]),mid+1,r);
else
{
update(a,mid,nw(son[k][0]),l,mid);
update(mid+1,b,nw(son[k][1]),mid+1,r);
}
sum[k]+=b+1-a;
}
}
long long query(int a,int b,int k,int l,int r)
{
if(a==l && b==r)return sum[k];
else
{
long long res=0;
int mid=(l+r)>>1;
if(b<=mid)res=query(a,b,nw(son[k][0]),l,mid);
else if(a>mid)res=query(a,b,nw(son[k][1]),mid+1,r);
else
{
res=query(a,mid,nw(son[k][0]),l,mid)+query(mid+1,b,nw(son[k][1]),mid+1,r);
}
res+=1LL*(b+1-a)*chg[k];
return res;
}
}
}
namespace sgt1
{
int root[1000010],cnt,rt,son[1000010][2];
void build(int &x,int l,int r)
{
x=++cnt;
if(l==r)return;
int mid=(l+r)>>1;
build(son[x][0],l,mid);
build(son[x][1],mid+1,r);
}
void update(int x,int l,int r,int v,int L,int R)
{
sgt2::update(L,R,sgt2::nw(root[x]),1,n);
if(l==r)return;
int mid=(l+r)>>1;
if(v<=mid)update(son[x][0],l,mid,v,L,R);
else update(son[x][1],mid+1,r,v,L,R);
}
int query(int x,int l,int r,long long v,int L,int R)
{
if(l==r)return l;
long long res=sgt2::query(L,R,sgt2::nw(root[son[x][1]]),1,n);
int mid=(l+r)>>1;
if(res>=v)return query(son[x][1],mid+1,r,v,L,R);
else return query(son[x][0],l,mid,v-res,L,R);
}
}
int main()
{
scanf("%d%d",&n,&m);
sgt1::build(sgt1::rt,1,n);
while(m--)
{
int op;
scanf("%d",&op);
if(op==1)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
sgt1::update(sgt1::rt,1,n,c,a,b);
}
else
{
int a,b;
long long c;
scanf("%d%d%lld",&a,&b,&c);
printf("%d\n",sgt1::query(sgt1::rt,1,n,c,a,b));
}
}
return 0;
}