3339: Rmq Problem/3585: mex 莫队算法

莫队大法好。
如果一个数>=n显然不可能成为答案。。所以根本不用写什么线段树。。直接上莫队就行了。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
int n,m,ans;
int c[200005],sum[200005],belong[200005];
struct node {int l,r,id,ans;} a[200005];
inline int read()
{
    int a=0,f=1; char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
    return a*f;
}
inline bool cmp(node a,node b)
{
    return belong[a.l]==belong[b.l]?a.r<b.r:belong[a.l]<belong[b.l];
}
inline bool cmp0(node a,node b)
{
    return a.id<b.id;
}
inline void update(int x,int val)
{
    if (x>=n) return;
    sum[x]+=val;
    if (sum[x]==0) ans=min(ans,x);
    if (sum[x]==1&&ans==x) 
        for (int i=x+1;;i++) if (!sum[i]) {ans=i; break;}
}
int main()
{
    n=read(); m=read();
    for (int i=1;i<=n;i++) c[i]=read();
    int block=(int)(sqrt(n));
    for (int i=1;i<=n;i++) belong[i]=(i-1)/block+1;
    for (int i=1;i<=m;i++) a[i].l=read(),a[i].r=read(),a[i].id=i;
    sort(a+1,a+m+1,cmp);
    int l=1,r=0;
    for (int i=1;i<=m;i++)
    {
        for (;r<a[i].r;r++) update(c[r+1],1);
        for (;r>a[i].r;r--) update(c[r],-1);
        for (;l<a[i].l;l++) update(c[l],-1);
        for (;l>a[i].l;l--) update(c[l-1],1);
        a[i].ans=ans;
    }
    sort(a+1,a+m+1,cmp0);
    for (int i=1;i<=m;i++) printf("%d\n",a[i].ans);
    return 0;
}