传送门
其实这个题大部分学过主席树的人肯定都会想到用主席树。
主席树做法就是记录每个颜色的出现次数,然后就可以做了,交换就swap一下,然后重建一下那个位置就好了。
然后我们可以算一下空间复杂度,
O(nlogn+m(opt2)logn)
,这样就会MLE
(比赛时只开了128MB)
我们可以写垃圾回收,然后又会TLE,所以主席树在128MB下还是做不出来这个题的。
我赛后提交的代码空间是140MB,刚好超一点。
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#define ll long long
using namespace std;
inline int read(){
int x=0;char ch=' ';int f=1;
while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
if(ch=='-')f=-1,ch=getchar();
while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
return x*f;
}
const int N=4e5+5;
struct node{
int ls,rs,size;
}t[N*30];
int n,m,tot,pos,ans;
int root[N],c[N];
inline void insert(int &now,int pre,int l,int r){
now=++tot;t[now]=t[pre];
if(l==r){t[now].size++;return;}
int mid=(l+r)>>1;
if(pos<=mid)insert(t[now].ls,t[pre].ls,l,mid);
else insert(t[now].rs,t[pre].rs,mid+1,r);
}
inline void query(int now,int pre,int l,int r){
if(l==r){ans=t[now].size-t[pre].size;return;}
int mid=(l+r)>>1;
if(pos<=mid)query(t[now].ls,t[pre].ls,l,mid);
else query(t[now].rs,t[pre].rs,mid+1,r);
}
int main(){
n=read();m=read();
for(int i=1;i<=n;i++)c[i]=read();
t[0].ls=t[0].rs=t[0].size=0;
for(int i=1;i<=n;i++)pos=c[i],insert(root[i],root[i-1],1,N-5);
while(m--){
int opt=read();
if(opt==1){
int l=read(),r=read();pos=read();
ans=0;
query(root[r],root[l-1],1,N-5);
printf("%d\n",ans);
}
else{
int x=read();
swap(c[x],c[x+1]);
pos=c[x];insert(root[x],root[x-1],1,N-5);
}
}
return 0;
}