1. 程式人生 > >BZOJ 3585: mex(分塊+莫隊)

BZOJ 3585: mex(分塊+莫隊)

傳送門

解題思路

  首先直接莫隊是能被卡的,時間複雜度不對。就考慮按照值域先進行分塊再進行莫隊,然後統計答案的時候就暴力掃所有的塊,直到一個塊內元素不滿,再暴力掃這個塊就行了,時間複雜度O(msqrt(n))

程式碼

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
 
using namespace std;
const int N=200005;
const int SIZ=600;
 
inline int rd(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)) f=ch=='-'?0:1,ch=getchar();
    while(isdigit(ch)) x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
    return f?x:-x;
}
 
int n,m,a[N],bl[N],siz,l[SIZ],r[SIZ],cnt[N],sum[SIZ],ans[N],num;
 
struct Query{
    int ql,qr,id;
    friend bool operator<(const Query A,const Query B){
        if(A.ql/siz!=B.ql/siz) return A.ql<B.ql;
        if((A.ql/siz)&1) return A.qr>B.qr;
        return A.qr<B.qr;
    }
}q[N];
 
inline void del(int x){
    if(a[x]>n) return;
    cnt[a[x]]--;if(!cnt[a[x]]) sum[bl[a[x]]]--;
}
inline void add(int x){
    if(a[x]>n) return ;
    if(!cnt[a[x]]) sum[bl[a[x]]]++;cnt[a[x]]++;
}
inline int query(){
    if(!cnt[0]) return 0;int pos=-1;
    for(int i=1;i<=num;i++)
        if(sum[i]!=r[i]-l[i]+1) {pos=i;break;}
    if(pos==-1) return n+1;
    for(int i=l[pos];i<=r[pos];i++)
        if(!cnt[i]) return i;
}
 
int main(){
    n=rd(),m=rd();siz=sqrt(n)+1;num=n/siz+(n%siz!=0);
    for(int i=1;i<=n;i++) a[i]=rd(),bl[i]=(i-1)/siz+1;
    for(int i=1;i<=num;i++) l[i]=(i-1)*siz+1,r[i]=i*siz;r[num]=n;
    for(int i=1;i<=m;i++) q[i].ql=rd(),q[i].qr=rd(),q[i].id=i;
    sort(q+1,q+1+m);int L=1,R=0;
    for(int i=1;i<=m;i++){
        while(L<q[i].ql) {del(L);L++;}
        while(L>q[i].ql) {L--;add(L);}
        while(R<q[i].qr) {R++;add(R);}
        while(R>q[i].qr) {del(R);R--;}
        ans[q[i].id]=query();
    }
    for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
    return 0;
}