莫队算法 P1972 区间不同元素的个数

题目链接：https://www.luogu.org/problemnew/show/P1972

就是区间不同值的个数。直接莫队模板。

#include
using namespace std;
typedef long long LL;
//由块号寻找第一个块元素的下标
#define LL(x) ((x-1)*Len+1)

const int maxn=1<<20;

struct node
{
    int L, R, k;
}q[maxn];

int pos[maxn];
LL ans[maxn];
LL flag[maxn];//保存当前区间前缀和的个数
int a[maxn];

bool cmp(node a, node b){

    return pos[a.L]==pos[b.L]?(pos[a.L]&1)?a.R<b.R:a.R>b.R:pos[a.L]<pos[b.L];
}

//bool cmp(node a, node b)
//{
//    if(pos[a.L]==pos[b.L])
//    {
//        return a.R
//    }
//
//    return pos[a.L]
//}

int n, m, k, Len;
int L=1, R=0;
LL ANS=0;

void add(int x)
{
    ANS+=(++flag[a[x]]==1);
//    flag[a[x]]++;
//    if(flag[a[x]]==1)
//    {
//        ANS++;
//    }
}

void del(int x)
{
    ANS-=(--flag[a[x]]==0);
//    flag[a[x]]--;
//    if(flag[a[x]]==0)
//    {
//        ANS--;
//    }

}

int main()
{
    scanf("%d",&n);
    Len=sqrt(n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        pos[i]=i/Len;
    }

    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&q[i].L,&q[i].R);
        q[i].k=i;
    }

    sort(q+1, q+1+m, cmp);
    flag[0]=1;//为了统计a[i]=k

    for(int i=1;i<=m;i++)
    {
        while(L<q[i].L)
        {
            del(L);
            L++;
        }
        while(L>q[i].L)
        {
            L--;
            add(L);
        }
        while(R<q[i].R)
        {
            R++;
            add(R);
        }
        while(R>q[i].R)
        {
            del(R);
            R--;
        }
        ans[q[i].k]=ANS;
    }
    for(int i=1;i<=m;i++)
    {
        printf("%d\n",ans[i]);
    }

    return 0;
}