洛谷 P1972 [SDOI2009]HH的项链（离线树状数组维护区间种类数）

传送门

---

题目大意

给出一个序列，现在有很多询问，每次询问一个区间内的不同种类数的个数。

解题思路

首先树状数组的题应该想到这两张图片：

假设给出的颜色序列为$1,1,1,3,2,4,3$，手玩一下查询区间，能够发现一个十分重要的性质：当$r$固定时，左边出现了重复的数，会产生相同的贡献，为了避免重复每次只考虑计算最靠右的数的贡献。那么需要记录每个数从左向右更新时记录下之前出现的位置，然后$update(pre\_pos,-1)$消除之前的贡献

例如$1,1,1,3$，假设当前部分查询的右界都为$4$，那么一开始更新时，树状数组的大小变化为（联系上面的图，实际上后缀的变化是和二进制的每一位相关的，但是查询前缀和与这一点没有关系）：

• 第一个数为$1$，之前没有出现，那么树状数组为$1,0,0,0$
• 第二个数为$1$，之前记录出现的位置为$1$，那么树状数组为$0,1,0,0$
• 第三个数为$1$，之前记录出现的位置为$2$，那么树状数组为$0,0,1,0$
• 第四个数为$3$，之前没有出现，那么树状数组为$0,0,1,1$
• 得到最新的树状数组$0,0,1,1$后，查询$[1,4],[2,4],[3,4],[4,4]$这些区间时，显然都是正确的

因为查询的区间和每个数的位置有关，与数的大小无关，那么显然是按照对应的$id$建树。考虑到上面的性质，而且树状数组查询的是前缀和，那么先保存下所有的右端点，然后按升序排序，最后离线更新和查询。

实际上左端点以及没有了任何用处，排序时右端点相同也没必要对左端点进行升序/降序处理。我们只需要维护一个不断向右更新的指针，当然指针不能超过右端点排序后每个位置的右界，然后更新过一个右端点的前缀，当需要查询这个以这个右端点为右界的区间时，直接类似前缀和的查询$getSum(r)-getSum(l-1)$

总之思想还是很巧妙的，讲的思路不算清晰，但也是目前理解$BIT$的极限了

//
// Created by Happig on 2020/8/23
//
#include <bits/stdc++.h>
#include <unordered_map>
#include <unordered_set>

using namespace std;
#define fi first
#define se second
#define pb push_back
#define ins insert
#define Vector Point
#define lowbit(x) (x&(-x))
#define mkp(x, y) make_pair(x,y)
#define mem(a, x) memset(a,x,sizeof a);
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef pair<double, double> pdd;
const double eps = 1e-8;
const double pi = acos(-1.0);
const int inf = 0x3f3f3f3f;
const double dinf = 1e300;
const ll INF = 1e18;
const int Mod = 1e9 + 7;
const int maxn = 1e6 + 10;

struct node{
    int l,r,id;

    bool operator < (const node &p)const{
        return r<p.r;
    }
}p[maxn];

int a[maxn],t[maxn],pre[maxn],ans[maxn];
bool vis[maxn];
int n;

void update(int i,int k){
    while(i<=n){
        t[i]+=k;
        i+=lowbit(i);
    }
}

ll getSum(int i){
    ll ans=0;
    for(;i;i-=lowbit(i)){
        ans+=t[i];
    }
    return ans;
}

int main() {
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
    ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int m;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    cin>>m;
    for(int i=1;i<=m;i++){
        cin>>p[i].l>>p[i].r;
        p[i].id=i;
    }
    sort(p+1,p+1+m);
    int s=1;
    for(int i=1;i<=m;i++){
        while(s<=p[i].r){
            if(pre[a[s]]) update(pre[a[s]],-1);
            update(s,1);
            pre[a[s]]=s;
            s++;
        }
        ans[p[i].id]=getSum(p[i].r)-getSum(p[i].l-1);
    }
    for(int i=1;i<=m;i++) cout<<ans[i]<<"\n";
    return 0;
}