mNOIP 模拟赛 day1 T2 数颜色

传送门
其实这个题大部分学过主席树的人肯定都会想到用主席树。
主席树做法就是记录每个颜色的出现次数,然后就可以做了,交换就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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值