BZOJ 4552 [Tjoi2016&Heoi2016]排序 线段树的分裂和合并

时间:2023-03-09 14:46:17
BZOJ 4552  [Tjoi2016&Heoi2016]排序 线段树的分裂和合并

https://www.lydsy.com/JudgeOnline/problem.php?id=4552

https://blog.csdn.net/zawedx/article/details/51818475

区间排序,这道题需要写两个线段树还要维护一个链表,有些细节,对我目前的代码能力来说有点算是码农题,但是理解思路之后调试起来出乎意料地简单。

这个写法的复杂度据说是nlogn的,我也不是很会算,反正能过就行(bushi)。

 #include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
#define LL long long
#define lc x*2
#define rc x*2+1
#define mid (l+r)/2
const int maxn=;
const int maxm=;
int n,m;
struct sgtree{
int rn[maxm];
void updata(int x){rn[x]=(rn[rc]?rn[rc]:rn[lc]);}
void ins(int x,int l,int r,int k,int v){
if(l==r){rn[x]=v;return;}
if(k>mid)ins(rc,mid+,r,k,v);
else ins(lc,l,mid,k,v);
updata(x);
}
int ask(int x,int l,int r,int k){
if(!rn[x])return ;
if(l==r)return rn[x];
int zz=;
if(mid<k)zz=ask(rc,mid+,r,k);
if(zz)return zz;
if(mid<=k)return rn[lc];
else return ask(lc,l,mid,k);
}
}e;
struct nod{ int l,r,tp,nxt; }c[maxm];
struct node{ int l,r,sum; }t[maxm];
queue<int>q;
inline int newp(){
int z=q.front();q.pop();
return z;
}
inline void delp(int x){
q.push(x);t[x]=t[];
}
void build(int x,int l,int r,int k){
t[x].sum=;
if(l==r)return;
if(k>mid)build(t[x].r=newp(),mid+,r,k);
else build(t[x].l=newp(),l,mid,k);
}
int mmerg(int x,int y){
if((!x)||(!y))return x+y;
t[x].l=mmerg(t[x].l,t[y].l);
t[x].r=mmerg(t[x].r,t[y].r);
t[x].sum=t[x].sum+t[y].sum;
delp(y); return x;
}
void mspli(int x,int y,int k){
int z=t[t[x].l].sum;
if(z<k)mspli(t[x].r,t[y].r=newp(),k-z);else swap(t[x].r,t[y].r);
if(z>k)mspli(t[x].l,t[y].l=newp(),k);
t[y].sum=t[x].sum-k;
t[x].sum=k;
}
int mask(int x,int l,int r,int k){
if(l==r)return l;
int z=t[t[x].l].sum;
if(k>z)return mask(t[x].r,mid+,r,k-z);
else return mask(t[x].l,l,mid,k);
}
int main(){
int x,v,op,l,r;
for(int i=;i<=maxm-;++i)q.push(i);
scanf("%d%d",&n,&m);
for(int i=,las=maxm-;i<=n;++i,las=x){
scanf("%d",&v);
c[las].nxt=x=newp();
c[x]=(nod){i,i,,};
e.ins(,,n,i,x);build(x,,n,v);
}
for(int i=;i<=m;++i){
scanf("%d%d%d",&op,&l,&r);
/*for(x=1;x<=n;++x){
int w=e.ask(1,1,n,x);
//cout<<w<<' ';
if(c[w].tp)printf("%d ",mask(w,1,n,c[w].r-x+1));
else printf("%d ",mask(w,1,n,x-c[w].l+1));
}printf("\n");*/
int lef=e.ask(,,n,l),rig,now,nxt;
if(l==c[lef].l){
rig=newp();
swap(t[rig],t[lef]); swap(c[rig],c[lef]);
c[now=lef]=(nod){l,r,op,rig};
}
else{
c[now=newp()]=(nod){l,r,op,rig=newp()};
if(!c[lef].tp) mspli(lef,rig,l-c[lef].l);
else {mspli(lef,rig,c[lef].r-l+);swap(t[lef],t[rig]);}
c[rig]=c[lef];c[rig].l=l;
c[lef].r=l-;c[lef].nxt=now;
}
for(nxt=rig;nxt&&r>=c[nxt].r;){
//cout<<t[nxt].sum<<endl;
mmerg(now,nxt);e.ins(,,n,c[nxt].l,);
int qq=nxt;nxt=c[nxt].nxt;c[qq]=c[];
}
c[now].nxt=nxt;
if(nxt!=&&c[nxt].l<=r){
e.ins(,,n,c[nxt].l,);
int zz=newp();
if(!c[nxt].tp) {mspli(nxt,zz,r-c[nxt].l+);swap(t[nxt],t[zz]);}
else mspli(nxt,zz,c[nxt].r-r);
mmerg(now,zz);
c[nxt].l=r+;
e.ins(,,n,r+,nxt);
}
e.ins(,,n,l,now);
}
/*for(x=1;x<=n;++x){
int w=e.ask(1,1,n,x);
//cout<<w<<' ';
if(c[w].tp)printf("%d ",mask(w,1,n,c[w].r-x+1));
else printf("%d ",mask(w,1,n,x-c[w].l+1));
}printf("\n");*/
scanf("%d",&x);
int w=e.ask(,,n,x);
if(c[w].tp)printf("%d\n",mask(w,,n,c[w].r-x+));
else printf("%d\n",mask(w,,n,x-c[w].l+));
return ;
}