BZOJ[2743][HEOI2012]采花 树状数组

题目链接

因为一段区间必须需要有两个颜色相同的点才有贡献
换句话说，只有第二个点才会有贡献

考虑离线，按询问按左端点从小到大将排序
用$nex_i$表示$i$号点下一个相同颜色点的位置，$fir_i$表示$i$号颜色第一个出现的位置

开一个树状数组$a$
先将每个$nex_{fir[i]}$在树状数组上+1
从左到右移动指针$l$，同时处理所有左端点为$i$的询问，统计$l \to r$的区间和
退出这个点的时候清除这个点的贡献，即将$a[nex_l]-1$，$a[nex_{nex[l]}]+1$即可

代码如下：

#include<algorithm>
#include<ctype.h>
#include<cstdio>
#define N 1000020
using namespace std;
inline int read(){
    int x=0,f=1;char c;
    do c=getchar(),f=c=='-'?-1:f; while(!isdigit(c));
    do x=(x<<3)+(x<<1)+c-'0',c=getchar(); while(isdigit(c));
    return x*f;
}
struct Query{
    int x,y,id;
}q[N];
inline bool cmp(Query a,Query b){
    return a.x<b.x;
}
int n,m,c;
int fir[N],nex[N],a[N],ans[N],p[N],top;
inline void add(int x,int v){
    for(;x<=n;x+=(x&(-x)))
        p[x]+=v;
}
inline int sum(int x){
    int tmp=0;
    for(;x;x-=(x&(-x)))
        tmp+=p[x];
    return tmp;
}
int main(){
    n=read();c=read(),m=read();
    for(int i=1;i<=n;i++)
        a[i]=read();
    for(int i=1;i<=m;i++){
        q[i].x=read();q[i].y=read();
        q[i].id=i;
    }
    sort(q+1,q+1+m,cmp);
    for(int i=n;i>=1;i--){
        nex[i]=fir[a[i]];///链表，记录每个点后面的第一个和它颜色相同的点的位置
        fir[a[i]]=i;
    }
    for(int i=1;i<=c;i++)
        if(nex[fir[i]])
            add(nex[fir[i]],1);
    for(int i=1;i<=n;i++){
        while(q[top+1].x==i){
            ans[q[top+1].id]=sum(q[top+1].y);
            top++;
        }
        if(nex[i]){
            add(nex[i],-1);
            if(nex[nex[i]]) add(nex[nex[i]],1);
        }
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
    return 0;
}