bzoj1878 [SDOI2009]HH的项链

题目链接

有点打脑壳

预处理每个点之后第一个相同颜色珠子出现的位置

然后将询问按左端点排序！

从左到右扫用树状数组维护前缀：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<ctime>
#include<queue>
#include<stack>
#include<map>
#include<set>
#define lowbit(x) x&-x;
using namespace std;
int getint()
{
    int ret=0;
    char ch=getchar();
    while(ch<'0'||ch>'9')ch=getchar();
    while(ch>='0'&&ch<='9')ret*=10,ret+=ch-'0',ch=getchar();
    return ret;
}
int n,m,Max;
int a[50050],next[50050],sum[50050],p[1000050];
struct wen
{
    int l,r,id,ans;
    bool operator < (const wen &a)const
    {
        return id<a.id;
    }
}q[200050];
bool com(const wen &a,const wen &b)
{
    return a.l==b.l?a.r<b.r:a.l<b.l;
}
void add(int i,int x)
{
    while(i<=n)sum[i]+=x,i+=lowbit(i);
}
int query(int i)
{
    int ret=0;
    while(i)ret+=sum[i],i-=lowbit(i);
    return ret;
}
int main()
{
    n=getint();
    for(int i=1;i<=n;i++)
    {
        a[i]=getint();
        Max=max(Max,a[i]);
    }
    for(int i=n;i>=1;i--)
        next[i]=p[a[i]],p[a[i]]=i;
    for(int i=1;i<=Max;i++)if(p[i])add(p[i],1);
    m=getint();
    for(int i=1;i<=m;i++)
    {
        q[i].l=getint(),q[i].r=getint();
        q[i].id=i;
    }
    sort(q+1,q+m+1,com);
    int l=1;
    for(int i=1;i<=m;i++)
    {
        while(l<q[i].l)
            if(next[l])add(next[l++],1);
            else l++;
        q[i].ans=query(q[i].r)-query(q[i].l-1);
    }
    sort(q+1,q+m+1);
    for(int i=1;i<=m;i++)
        printf("%d
",q[i].ans);
    return 0;
}