【SPOJ DQUERY - D-query】 莫队算法

https://www.spoj.com/problems/DQUERY/en/
题意就是区间不同数的个数，模板题
代码

#include<stdio.h>
#include<math.h>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 1e6+5;

struct data
{
    int l,r,id;
}Q[maxn];
long long ans[maxn];
int pos[maxn];
int a[maxn];
bool cmp(const data &a,const data &b)
{
    if(pos[a.l]==pos[b.l]) return a.r<b.r;
    return pos[a.l]<pos[b.l];
}
int flag[maxn];
int L=1,R=0;
long long  Ans=0;
void add(int x)
{
    flag[a[x]]++;
    if(flag[a[x]]==1) Ans++;
}

void del(int x)
{
    flag[a[x]]--;
    if(flag[a[x]]==0) Ans--;
}
int main()
{
    int n,m;
    scanf("%d",&n);
    int sz=sqrt(n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        pos[i]=i/sz;
    }
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&Q[i].l,&Q[i].r);
        Q[i].id=i;
    }
    sort(Q+1,Q+1+m,cmp);
    for(int i=1;i<=m;i++)
    {
        while(R<Q[i].r)
        {
            R++;
            add(R);
        }
        while(L>Q[i].l)
        {
            L--;
            add(L);
        }
        while(L<Q[i].l)
        {
            del(L);
            L++;
        }
        while(R>Q[i].r)
        {
            del(R);
            R--;
        }
        ans[Q[i].id]=Ans;
    }
    for(int i=1;i<=m;i++)
        printf("%lld\n",ans[i]);
    return 0;
}

友情推荐莫队算法专题链接