【bzoj2743】[HEOI2012]采花

树状数组

#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
using namespace std;

#define N 1000010

#define lowbit(x) (x & (-x))

int n,c,m;

int t[N],next[N],ans[N],p[N],a[N];

struct Node
{
       int l,r,id;
}q[N];

int cmp(Node a,Node b)
{
    return a.l<b.l;
}

void update(int x,int d)
{
     while (x<=n)
           t[x]+=d,x+=lowbit(x);
}

int ask(int x)
{
    int res(0);
    while (x>0)
          res+=t[x],x-=lowbit(x);
    return res;
}

int main()
{
    scanf("%d%d%d",&n,&c,&m);
    for (int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for (int i=n;i>0;i--)
        next[i]=p[a[i]],p[a[i]]=i;
    for (int i=1;i<=c;i++)
        if (next[p[i]])
           update(next[p[i]],1);
    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 l=1;
    for (int i=1;i<=m;i++)
    {
        while (l<q[i].l)
        {
              if (next[l])
                 update(next[l],-1);
              if (next[next[l]])
                 update(next[next[l]],1);
              l++;
        }
        ans[q[i].id]=ask(q[i].r)-ask(q[i].l-1);
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
    return 0;
}