bzoj2743: [HEOI2012]採花（樹狀陣列）

題目傳送門
。

解法：
想了蠻久。

首先這道題肯定不能用字首和相減什麼的。
那麼只能用區間1~l來表示答案。
因為1~r就沒有了l的限制。

記錄每個點上一個出現的位置。
last[i]表示i上一個出現的位置。
能採到i這個位置上的花是從last[last[i]]+1這個位置開始的。
然後last[i]+1開始就採不了這朵花了。

樹狀陣列維護下

程式碼實現：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
int lowbit(int x) {return x&-x;}
int n,s[1100000],a[1100000];
void change(int x,int c) {while(x<=n) {s[x]+=c;x+=lowbit(x);}}
int find_sum(int x) {
    int ans=0;
    while(x>0){ans+=s[x];x-=lowbit(x);}
    return ans;
}
int last[1100000],next[1100000];
struct node {int l,r,id,ans;}Q[1100000];
bool cmp(node n1,node n2) {return n1.r<n2.r;}
bool cmp1(node n1,node n2) {return n1.id<n2.id;}
int main() {
    int c,m;scanf("%d%d%d",&n,&c,&m);
    memset(last,0,sizeof(last));memset(next,0,sizeof(next));
    for(int i=1;i<=n;i++) {
        scanf("%d",&a[i]);last[i]=next[a[i]];next[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+1+m,cmp);int now=0;memset(s,0,sizeof(s));
    for(int i=1;i<=m;i++) {
        while(now<Q[i].r) {
            now++;
            if(last[now]!=0) {
                change(last[last[now]]+1,1);
                change(last[now]+1,-1);
            }
        }Q[i].ans=find_sum(Q[i].l);
    }sort(Q+1,Q+1+m,cmp1);
    for(int i=1;i<=m;i++)printf("%d\n",Q[i].ans);
    return 0;
}