luogu P3901 數列找不同

題目描述

現有數列A1,A2,⋯,A N，Q 個詢問(L i,R i)，ALi,ALi+1,⋯,ARi 是否互不相同

輸入格式

第1 行，2 個整數N,Q

第2 行，N 個整數ALi,ALi+1,⋯,A R i

Q 行，每行2 個整數L i,R i

輸出格式

對每個詢問輸出一行，“Yes” 或者“No”

---

怎麼判區間內的數是否互不相同呢？

設區間長度爲len，互不相同可以看成是區間內有len個不同的數。那麼我們統計就好了。

暴力一點，統計用莫隊

時間複雜度爲O(N√N)

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#define maxn 100001
using namespace std;
int n,m,val[maxn];
bool ans[maxn];
inline int read(){
    register int x(0),f(1); register char c(getchar());
    while(c<'0'||'9'<c){ if(c=='-') f=-1; c=getchar(); }
    while('0'<=c&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar();
    return x*f;
}

struct query{
    int col,l,r,id;
    bool operator<(const query &q)const{ return col==q.col?r<q.r:l<q.l; }
}q[maxn];
int unit,tot,cnt[maxn];
inline void del(int x){ tot-=(--cnt[val[x]]==0); }
inline void add(int x){ tot+=(++cnt[val[x]]==1); }
inline void solve(){
    sort(q+1,q+1+m);
    int l=0,r=0;
    for(register int i=1;i<=m;i++){
        while(l<q[i].l) del(l++);
        while(l>q[i].l) add(--l);
        while(r<q[i].r) add(++r);
        while(r>q[i].r) del(r--);
        ans[q[i].id]=(tot==q[i].r-q[i].l+1);
    }
}

int main(){
    n=read(),m=read(),unit=sqrt(n);
    for(register int i=1;i<=n;i++) val[i]=read();
    for(register int i=1;i<=m;i++) q[i].l=read(),q[i].r=read(),q[i].id=i,q[i].col=q[i].l/unit+1;
    solve();
    for(register int i=1;i<=m;i++) printf("%s\n",ans[i]?"Yes":"No");
    return 0;
}