1. 程式人生 > >BZOJ 3339 && luogu4137 Rmq Problem / mex(莫隊)

BZOJ 3339 && luogu4137 Rmq Problem / mex(莫隊)

答案 自然 輸入輸出格式 getch struct problem getchar() 輸出 格式

P4137 Rmq Problem / mex

題目描述

有一個長度為n的數組{a1,a2,…,an}。m次詢問,每次詢問一個區間內最小沒有出現過的自然數。

輸入輸出格式

輸入格式:

第一行n,m。
第二行為n個數。
從第三行開始,每行一個詢問l,r。

輸出格式:

一行一個數,表示每個詢問的答案。

輸入輸出樣例

輸入樣例#1:
5 5
2 1 0 2 1
3 3
2 3
2 4
1 2
3 5

輸出樣例#1:

1
2
3
0
3

說明

對於30%的數據:1<=n,m<=1000
對於100%的數據:1<=n,m<=200000,0<=ai<=10^9,1<=l<=r<=n

簡單莫隊

不過這個轉換不一定是O(n)的

估計這道題卡莫隊很容易吧

老是細節爆炸,真的不知道怎嘛辦

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <cstdio>
#define R register
#define maxn 301000
using namespace std;
int n,m,now;
int a[maxn];
int belong[maxn];
int vis[maxn];
int ma;
struct node {
    int l,r,id;
} q[maxn];
int ans[maxn];
inline int read()
{
    int x=0,f=1;char s=getchar();
    while('0'>s||s>'9') {
        if(s=='-') f=-1;
        s=getchar();
    }
    while('0'<=s&&s<='9') {
        x=x*10+s-'0';
        s=getchar();
    }
    return x*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 void add(int x) {
    ++vis[x];
    if(now < x) return;
    if(now==x&&vis[x]==1) { //細節在這裏
        for(R int i=now+1; i<n; ++i) {
            if(!vis[i]) {
                now=i;
                return;
            }
        }
    }
}
inline void delet(int x) {
    --vis[x];
    if(now < x) return;
    if(now > x) {
        if(!vis[x]) {
            now=x;
        }
    }
}
int main() {
    n=read();
    m=read();
    int k=sqrt(n);
    //別看a[]很大,ans範圍小於n
    for(R int i=1; i<=n; ++i) {
        a[i]=read();
        if(a[i] > n)
            a[i]=n+1;
    }
    if(n<=1000&&m<=1000) { //baoli
        while(m--) {
            int l,r;
            memset(vis,0,sizeof(vis));
            l=read();
            r=read();
            for(R int i=l; i<=r; ++i)
                vis[a[i]]=1;
            R int i=0;
            while(vis[i]) i++;
            printf("%d\n",i);
        }
        return 0;
    }
 
    for(R int i=1; i<=n; ++i)
        belong[i]=(i-1)/k+1;
    for(R int i=1; i<=m; ++i)
    {
        q[i].l=read();
        q[i].r=read();
        q[i].id=i;  
    }
    sort(q+1,q+1+m,cmp);
    int l=1,r=0;
    for(R int i=1; i<=m; ++i) {
        while(l > q[i].l) add(a[--l]);
        while(l < q[i].l) delet(a[l++]);
        while(r < q[i].r) add(a[++r]);
        while(r > q[i].r) delet(a[r--]);
        ans[q[i].id]=now;
    }
    for(R int i=1; i<=m; ++i)
        printf("%d\n",ans[i]);
    return 0;
}

BZOJ 3339 && luogu4137 Rmq Problem / mex(莫隊)