Bzoj P1878 [SDOI2009]HH的项链___树状数组+前缀和

题目大意：

给出含N个数的一段序列，M个询问，每次询问一个区间[l,r]，回答[l,r]中不同的数有多少个。

数为编号为0到1000000之间的整数
1 ≤ L ≤ R ≤ N
N ≤ 50000，M ≤ 200000

题解：

以前对树状数组只是有些了解，
今天才终于弄明白了实质，
不是只记得x & (-x)的我了..
这题如果在线做，O(MN^2),显然吃不消..
我们考虑一下离线，
对于r我们升序排列，然后我们就可以用前缀和乱搞了
对于一个i位置的数j
我们可以将上一个j出现的位置k
[k+1,n]的位置全部+1,[i+1,n]全部-1
这样的话就可以保证每次查询不会有错漏
因为这样前缀和时间为O(NM)依然会超时…
所以我们加个树状数组优化一下就可以了
时间复杂度:(M log N)

代码：

#include<bits/stdc++.h>
#define N 200005

using namespace std ;

struct node{
    int l,r,cnt;
}a[N];
int next[N],ans[N],c[N],ls[1000005],n,m,rp;

int lowbit(int x){
    return x & (-x);
}

void insert(int x ,int y){
    if (x == n) return;
    while (x <= n){
        c[x] += y;
        x += lowbit(x);
    }
}
int getsum(int x){
    int sum = 0;
    while (x){
        sum += c[x];
        x -= lowbit(x);
    }
    return sum;
}

int cmp(node aa, node bb){
    return aa.r < bb.r;
}

int main(){
    scanf("%d",&n);
    int s;
    for (int i=1; i<=n; i++){
         scanf("%d",&s);
         next[i]=ls[s];
         ls[s]=i;
    }
    scanf("%d",&m) ;
    for (int i=1; i<=m; i++){
         scanf("%d%d",&a[i].l,&a[i].r);
         a[i].cnt=i;
    }
    sort(a+1,a+1+m,cmp);
    for (int i=1; i<=m; i++){
         while (rp < a[i].r){
                rp++;
                insert(next[rp]+1,1);
                insert(rp+1,-1);
         }
         ans[a[i].cnt]=getsum(a[i].l);
    }
    for (int i=1; i<=m; i++) printf("%d\n",ans[i]);
    return 0;
}