牛客网暑期ACM多校训练营（第一场）-优快云博客

本文链接：https://blog.youkuaiyun.com/qq_37129433/article/details/81132318

本文介绍了一种使用莫队算法求解特定区间内不同元素数量的方法。针对给定的数组，通过预处理和算法优化，实现了高效查询任意区间内不同整数的数量。适用于处理大量查询请求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意: 给你一个数值元素为1到n，大小为n的数组，q个询问每次输入[l,r]，求a1 a2
a3 … al,ar ar+1 ar+2…. an中不同元素的个数
输入范围:
1<=n ,q <= 1e5
1<= ai <=n
1 <= li,ri <=n
The number of test cases does not exceed 10.
输入样例:

输入样例:

2
1
3

思路：用莫队求一个区间[l,r]不同元素的个数的板子网上一大把。把原来区间扩大到两倍。套莫队模板。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
using namespace std;
const int N=1e5+5;//区间范围
const int MAX=1e5+5;//最大数字
int unit,cnt[MAX],arr[2*N],res[2*N],ans=0;

struct node {
    int l,r,id;
    int l1,r1;
} q[N];

bool cmp(node a,node b) {
    return a.l/unit!=b.l/unit?a.l/unit<b.l/unit:a.r<b.r;
}

void add(int pos) {
    cnt[arr[pos]]++;
    if(cnt[arr[pos]]==1) {
        ans++;
    }
}

void remove(int pos) {
    cnt[arr[pos]]--;
    if(cnt[arr[pos]]==0) {
        ans--;
    }
}

int main() {
    int n,m;
    while(~scanf("%d%d",&n,&m)) {
        ans= 0;
        memset(cnt,0,sizeof(cnt));
        memset(res,0,sizeof(res));
        unit=sqrt(n);
        for(int i=1; i<=n; i++) {
            scanf("%d",&arr[i]);
        }
        for(int i=n+1;i<=2*n;i++){
            arr[i] = arr[i-n];
        }

        for(int i=1; i<=m; i++) {
            scanf("%d%d",&q[i].l,&q[i].r);
            q[i].id=i;
            q[i].l = min(q[i].l,q[i].r);
            q[i].r = max(q[i].l,q[i].r);
            int temp = q[i].l;
            q[i].l = q[i].r;
            q[i].r = temp+n;
        }
        sort(q+1,q+m+1,cmp);
        int L=q[1].l,R=L-1;
        for(int i=1; i<=m; i++) {
            while(L > q[i].l)
                add(--L);
            while(L<q[i].l)
                remove(L++);
            while(R>q[i].r)
                remove(R--);
            while(R<q[i].r)
                add(++R);
            res[q[i].id]=ans;
        }

        for(int i=1; i<=m; i++) {
            printf("%d\n",res[i]);
        }
    }


}