树状数组维护区间种类数

[SDOI2009] HH的项链

题目描述

HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运，所以每次散步完后，他都会随意取出一段贝壳，思考它们所表达的含义。HH 不断地收集新的贝壳，因此，他的项链变得越来越长。

有一天，他突然提出了一个问题：某一段贝壳中，包含了多少种不同的贝壳？这个问题很难回答…… 因为项链实在是太长了。于是，他只好求助睿智的你，来解决这个问题。

输入格式

一行一个正整数 $n$，表示项链长度。
第二行 $n$ 个正整数 $a_i$，表示项链中第 $i$ 个贝壳的种类。

第三行一个整数 $m$，表示 HH 询问的个数。
接下来 $m$ 行，每行两个整数 $l,r$，表示询问的区间。

输出格式

输出 $m$ 行，每行一个整数，依次表示询问对应的答案。

样例 #1

样例输入 #1

6
1 2 3 4 3 5
3
1 2
3 5
2 6

样例输出 #1

2
2
4

提示

【数据范围】

对于 $20\%$ 的数据，$1\le n,m\le 5000$；
对于 $40\%$ 的数据，$1\le n,m\le 10^5$；
对于 $60\%$ 的数据，$1\le n,m\le 5\times 10^5$；
对于 $100\%$ 的数据，$1\le n,m,a_i\le 10^6$，$1\le l\le r\le n$。

思路：

将查询离线，使用last数组记录颜色 $a_i$ 上一次出现的位置，当遇到当前颜色，就增加当前点对于查询的贡献，并且消除 $last[a_i]$的影响即可。

#include <bits/stdc++.h>

using namespace std;
const int N = 2e6 + 5;
typedef long long ll;
typedef pair<ll, ll> pll;
typedef array<int, 3> ar;
int mod = 1e9+7;
const int maxv = 4e6 + 5;
#define endl "\n"


struct MIT
{
ll tr[N];
int lowbit(int x) {
    return x & (-x);
}

void add(int u, int v) {
    for (int i = u; i < N; i += lowbit(i)) {
        tr[i] += v;
    }
}

ll query(int x) {
    ll res = 0;

    for (int i = x; i > 0; i -= lowbit(i)) {
        res += tr[i];
    }

    return res;
}
};
MIT tr;
vector<pll> qy[N];

void solve()
{
    int n;
    cin>>n;
    vector<int> a(n+5);
    for(int i=1;i<=n;i++) cin>>a[i];
    int q;
    cin>>q;
    for(int i=1;i<=q;i++){
        int l,r;
        cin>>l>>r;
        qy[r].push_back({l,i});
    }
    vector<int> last(N);
    vector<int> ans(q+5);
    for(int i=1;i<=n;i++){
        if(!last[a[i]]){
            tr.add(i,1);
            last[a[i]]=i;
        }
        else{
            tr.add(last[a[i]],-1);
            last[a[i]]=i;
            tr.add(i,1);
        }
        for(auto [l,id]: qy[i]){
            ans[id]=tr.query(i)-tr.query(l-1);
        }
    }
    for(int i=1;i<=q;i++){
        cout<<ans[i]<<endl ;
    }
//     cout<<endl;
    
} 


int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    int t;
    t=1;
    // cin>>t;
    while(t--){
        solve();
    }
    system("pause");
    return 0;
}