洛谷 P1972 [SDOI2009]HH的项链

 

题目描述

HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运，所以每次散步完后，他都会随意取出一段贝壳，思考它们所表达的含义。HH 不断地收集新的贝壳，因此，他的项链变得越来越长。有一天，他突然提出了一个问题：某一段贝壳中，包含了多少种不同的贝壳？这个问题很难回答……因为项链实在是太长了。于是，他只好求助睿智的你，来解决这个问题。

输入输出格式

输入格式：

 

第一行：一个整数N，表示项链的长度。

第二行：N 个整数，表示依次表示项链中贝壳的编号（编号为0 到1000000 之间的整数）。

第三行：一个整数M，表示HH 询问的个数。

接下来M 行：每行两个整数，L 和R（1 ≤ L ≤ R ≤ N），表示询问的区间。

 

输出格式：

 

M 行，每行一个整数，依次表示询问对应的答案。

 

输入输出样例

输入样例#1： 复制

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

输出样例#1： 复制

2
2
4

说明

数据范围： 

对于100%的数据，N <= 500000，M <= 500000。

考虑离线+树状数组

 1.按r排序一下

2.树状数组维护c[i]从1到i不同数字的个数有多少个

3.前缀和的思想（c[r]-c[l-1]）

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxn=5e5+5;
int c[maxn],a[maxn],ans[maxn],n,m;
std::vector<pair<int,pair<int,int> > > q;
unordered_map<int,int>mp;
inline int lowbit(int x){return x&(-x);}
inline void update(int i,int val)
{
	while(i<=n){c[i]+=val;i+=lowbit(i);}
}
inline int query(int i)
{
	int res=0;while(i>0){res+=c[i];i-=lowbit(i);}return res;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
	scanf("%d",&m);
    for(int i=0;i<m;i++)
    {
    	int l,r;scanf("%d%d",&l,&r);
    	q.emplace_back(make_pair(r,make_pair(l,i)));//将r作为第一关键字，方便排序
    }
    sort(q.begin(),q.end());int cnt=1;
    for(int i=0;i<m;i++)
    {
    	while(cnt<=q[i].first)
    	{
    		if(mp[a[cnt]])update(mp[a[cnt]],-1);
    		update(mp[a[cnt]]=cnt,1);cnt++;
    	}
    	ans[q[i].second.second]=query(q[i].first)-query(q[i].second.first-1);
    }
    for(int i=0;i<m;i++)printf("%d\n",ans[i]);
	return 0;
}