2743: [HEOI2012]采花

题目链接

题目大意：给出一个数列，多次询问查询[L,R]中出现大于等于两次的数字种数

题解：莫队做法类似1878HH的项链，但是这题数据太大。莫队会T。记pre[i]为每个位置i的数字的前一个相同数字出现的位置，若不存在则pre[i]=0。然后就可以搞了，类似于1878HH的项链

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define M 1000005

int n,m,C;
int a[M],ans[M],c[M],pre[M],last[M];

struct node{int l,r,id;}q[M];

bool cmp(node x,node y){return x.r<y.r;}

void updata(int x,int v){
    for(int i=x;i<=n;i+=i&(-i))
        c[i]+=v;
}

int query(int x){
    int ret=0;
    for(int i=x;i>0;i-=i&(-i))
        ret+=c[i];
    return ret;
}

void work()
{
    int now=0;
    for(int i=1;i<=m;i++)
    {
        while(now<q[i].r) now++,updata(pre[pre[now]]+1,1),updata(pre[now]+1,-1);
        ans[q[i].id]=query(q[i].l);
    }
    for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
}

void init()
{
    cin>>n>>C>>m;
    for(int i=1,x;i<=n;i++)
        scanf("%d",&a[i]),pre[i]=last[a[i]],last[a[i]]=i;
    for(int i=1;i<=m;i++)
        scanf("%d%d",&q[i].l,&q[i].r),q[i].id=i;
    sort(q+1,q+m+1,cmp);
}

int main()
{
    init();
    work();
    return 0;
}