[SDOI2009] HH的项链
题目描述
HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运,所以每次散步完后,他都会随意取出一段贝壳,思考它们所表达的含义。HH 不断地收集新的贝壳,因此,他的项链变得越来越长。
有一天,他突然提出了一个问题:某一段贝壳中,包含了多少种不同的贝壳?这个问题很难回答…… 因为项链实在是太长了。于是,他只好求助睿智的你,来解决这个问题。
输入格式
一行一个正整数
n
n
n,表示项链长度。
第二行
n
n
n 个正整数
a
i
a_i
ai,表示项链中第
i
i
i 个贝壳的种类。
第三行一个整数
m
m
m,表示 HH 询问的个数。
接下来
m
m
m 行,每行两个整数
l
,
r
l,r
l,r,表示询问的区间。
输出格式
输出 m m m 行,每行一个整数,依次表示询问对应的答案。
样例 #1
样例输入 #1
6
1 2 3 4 3 5
3
1 2
3 5
2 6
样例输出 #1
2
2
4
提示
【数据范围】
对于
20
%
20\%
20% 的数据,
1
≤
n
,
m
≤
5000
1\le n,m\leq 5000
1≤n,m≤5000;
对于
40
%
40\%
40% 的数据,
1
≤
n
,
m
≤
1
0
5
1\le n,m\leq 10^5
1≤n,m≤105;
对于
60
%
60\%
60% 的数据,
1
≤
n
,
m
≤
5
×
1
0
5
1\le n,m\leq 5\times 10^5
1≤n,m≤5×105;
对于
100
%
100\%
100% 的数据,
1
≤
n
,
m
,
a
i
≤
1
0
6
1\le n,m,a_i \leq 10^6
1≤n,m,ai≤106,
1
≤
l
≤
r
≤
n
1\le l \le r \le n
1≤l≤r≤n。
思路:
将查询离线,使用last数组记录颜色 a i a_i ai 上一次出现的位置,当遇到当前颜色,就增加当前点对于查询的贡献,并且消除 l a s t [ a i ] last[a_i] last[ai]的影响即可。
#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;
}