*NOJ2446.小陈的气球(树状数组+区间处理+离线处理）

题目链接：https://acm.njupt.edu.cn/problem/NOJ2446
题意:有n个气球，每次把区间[L,R]之间的气球去掉，问还剩下多少中颜色的气球
解题思路:
①先将气球序列扩大两倍，记录下前i个气球中的种类数sum和每一个位置处气球对应相同颜色的下一个气球的位置Next
②离线处理询问，那么题目中的删除[L,R]区间内的气球，求[1,L-1]和[R+1,N]之间气球的颜色就可以转换为求[R+1,N+L-1]区间内的气球个数，同时将R+1看成L,N+L-1看作R，重新保存到结构体中
③将转换后的询问，按L从小到大的顺序进行排列，求[L,R]区间内气球的种类数：sum[R]-sum[L]+(区间[1,L-1]中的气球颜色和[L,R]之间的气球颜色重复的种类数）
④其中区间[1,L-1]中的气球颜色和[L,R]之间的气球颜色重复的种类数，使用的是树状数组进行维护，先令树状数组初值为0，然后将[1,L-1]之间的气球对应同颜色下一个位置的气球+1即可，最后利用getsum求值得到的就是重复的气球颜色数。因为询问是按照L从小到大的顺序，所以每次处理[perl,L-1]之间的气球颜色数就可以了

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<string.h>
#include<string>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn = 1e5 + 10;

struct node {
	int l, r;
	int pos;
	bool operator < (const node a)const {
		return l == a.l ? r < a.r : l < a.l;
	}
};

int n, t;
int a[2 * maxn];
int Next[2 * maxn];
int sum[2 * maxn];
int vis[2 * maxn];
int ans[maxn];
vector<node> v;


int c[2*maxn]; //对应原数组和树状数组

int lowbit(int x) {
	return x & (-x);
}

void updata(int i, int k) {    //在i位置加上k
	while (i <= 2 * n) {
		c[i] += k;
		i += lowbit(i);
	}
}

int getsum(int i) {        //求A[1 - i]的和
	int res = 0;
	while (i > 0) {
		res += c[i];
		i -= lowbit(i);
	}
	return res;
}

int main() {
	cin >> n >> t;
	for (int i = 1; i <= n; i++) {
		scanf("%d", &a[i]);
		a[n + i] = a[i];
	}
	for (int i = 1; i <= 2 * n; i++) {
		if (vis[a[i]] == 0) {
			sum[i] = sum[i - 1] + 1;
			vis[a[i]] = 1;
		}
		else {
			sum[i] = sum[i - 1];
		}
	}
	memset(vis, -1, sizeof(vis));
	for (int i = 2 * n; i >= 1; i--) {
		Next[i] = vis[a[i]];
		vis[a[i]] = i;
	}
	for (int i = 1; i <= t; i++) {
		int L, R;
		scanf("%d%d", &L, &R);
		node tmp; tmp.l = R + 1; tmp.r = L + n - 1; tmp.pos = i;
		v.push_back(tmp);
	}
	sort(v.begin(), v.end());
	int prel = 1;
	for (int i = 0; i < t; i++) {
		node tmp = v[i];
		ans[tmp.pos] = sum[tmp.r] - sum[tmp.l - 1];
		for (int j = prel; j <= tmp.l - 1; j++) {
			if (Next[j] == -1)
				continue;
			updata(Next[j],1);
		}
		ans[tmp.pos] += getsum(tmp.r) - getsum(tmp.l-1);
		prel= tmp.l;
	}
	for (int i = 1; i <= t; i++) {
		printf("%d\n", ans[i]);
	}
	return 0;
}