[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;
}