HH的项链 洛谷P1972 树状数组

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

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

输入格式

一行一个正整数 nn，表示项链长度。
第二行 nn 个正整数 a_iai​，表示项链中第 ii 个贝壳的种类。

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

输出格式

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

输入输出样例

输入 #1复制

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

输出 #1复制

2
2
4

说明/提示

【数据范围】

对于 20\%20% 的数据，1\le n,m\leq 50001≤n,m≤5000；
对于 40\%40% 的数据，1\le n,m\leq 10^51≤n,m≤105；
对于 60\%60% 的数据，1\le n,m\leq 5\times 10^51≤n,m≤5×105；
对于 100\%100% 的数据，1\le n,m,a_i \leq 10^61≤n,m,ai​≤106，1\le l \le r \le n1≤l≤r≤n。

本题可能需要较快的读入方式，最大数据点读入数据约 20MB

思路：使用树状数组 树状数组可以查询1到r的一个区间值（不同数字出现的次数） 这里需要忽略那些重复出现的数字（一个数字只出现一次）即删除在前面已经加入树状数组的数字然后把当前数字加入树状数组 如果这个数字前面没有出现那么直接加入树状数组 那么查询l到r就是用1到r的值减去1到l-1的值 所以我们可以按以右端点从小到大为顺序遍历修改树状数组 对于一个询问 修改的区间为上一个r+1到当前r 

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1000010;
typedef long long ll;
int a[MAXN],ans[MAXN],c[MAXN],vis[MAXN],n;
int lowbit(int x)  //求2^k k为x末尾1的位置 从零位开始
{
	return x&(-x);
}
void update(int x,int k) //修改原数组的值并修改树状数组的值
{
	while(x <= n){
		c[x] += k;
		x += lowbit(x);
	}
}
ll getsum(int x) //求前缀和 c数组代表树状数组
{
	ll s = 0;
	while(x > 0){
		s += c[x];
		x -= lowbit(x);
	}
	return s;
}
struct info{
	int l,r,num;
	bool operator < (const info & a) const{
		return r < a.r;
	}
}node[MAXN]; 
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	cin >> n;
	for(int i = 1;i <= n; i++) cin >> a[i];
	int m;
	cin >> m;
	for(int i = 1;i <= m; i++){
		cin >> node[i].l >> node[i].r;
		node[i].num = i;
	}
	sort(node+1,node+1+m);
	int be = 1;
	for(int i = 1;i <= m; i++){
		for(int j = be;j <= node[i].r; j++){
			if(vis[a[j]] > 0) update(vis[a[j]],-1);
			update(j,1);
			vis[a[j]] = j;//记录出现的位置
		}
		be = node[i].r+1;//下次修改的起始位置
		ans[node[i].num] = getsum(node[i].r) - getsum(node[i].l-1);
	}
	for(int i = 1;i <= m; i++){
		cout << ans[i] << "\n";
	}
	return 0;
}