【BZOJ】2743: [HEOI2012]采花（树状数组）

题目

传送门：QWQ

分析

已经凉凉。看错数据范围敲了发莫队........

和HH的项链差不多，把每种颜色之前的颜色到再之前的颜色这段区间 区间加。

区间加就树状数组特技

代码

#include <bits/stdc++.h>
const int maxn=2e6+;
using namespace std;
struct data{
    int num,l,r;
}Q[maxn];
int c[maxn],ans[maxn],pre[maxn],front[maxn],bit[maxn];
int n,m,o;
bool cmp(data a,data b){
    return a.r<b.r;
}
void add(int x,int a){if(!x)return; for(;x<=n;x+=x&-x) bit[x]+=a; }
int sum(int x){int ans=; for(;x>;x-=x&-x) ans+=bit[x]; return ans;}
int main(){
     scanf("%d%d%d",&n,&o,&m);
    for(int i=;i<=n;i++) scanf("%d",&c[i]);
    for(int i=;i<=m;i++){
        scanf("%d%d",&Q[i].l,&Q[i].r);
        Q[i].num=i;
    }
    sort(Q+,Q++m,cmp);
    int now=;
    for(int i=;i<=m;i++){
        pre[i]=front[c[i]]; front[c[i]]=i;
        add(pre[pre[i]],-); add(pre[i],);
        while(Q[now].r==i) ans[Q[now].num]=sum(i)-sum(Q[now].l-),now++;
    }
    for(int i=;i<=m;i++){
        printf("%d\n",ans[i]);
    }
    return ;
}