2743: [HEOI2012]采花（离线树状数组）

2743: [HEOI2012]采花

Time Limit: 15 Sec   Memory Limit: 128 MB
Submit: 2365   Solved: 1218
[ Submit][ Status][ Discuss]

Description

萧芸斓是Z国的公主，平时的一大爱好是采花。

今天天气晴朗，阳光明媚，公主清晨便去了皇宫中新建的花园采花。花园足够大，容纳了n朵花，花有c种颜色（用整数1-c表示），且花是排成一排的，以便于公主采花。公主每次采花后会统计采到的花的颜色数，颜色数越多她会越高兴！同时，她有一癖好，她不允许最后自己采到的花中，某一颜色的花只有一朵。为此，公主每采一朵花，要么此前已采到此颜色的花，要么有相当正确的直觉告诉她，她必能再次采到此颜色的花。由于时间关系，公主只能走过花园连续的一段进行采花，便让女仆福涵洁安排行程。福涵洁综合各种因素拟定了m个行程，然后一一向你询问公主能采到多少朵花（她知道你是编程高手，定能快速给出答案！），最后会选择令公主最高兴的行程（为了拿到更多奖金！）。

Input

 第一行四个空格隔开的整数n、c以及m。接下来一行n个空格隔开的整数，每个数在[1, c]间，第i个数表示第i朵花的颜色。接下来m行每行两个空格隔开的整数l和r（l ≤ r），表示女仆安排的行程为公主经过第l到第r朵花进行采花。

Output

共m行，每行一个整数，第i个数表示公主在女仆的第i个行程中能采到的花的颜色数。

Sample Input

5 3 5

1 2 2 3 1

1 5

1 2

2 2

2 3

3 5

Sample Output

2

0 0 1 0

处理方式和 Codeforces Round #426 (Div. 1)：B. The Bakery 这题很像

先将询问按右端点排序，之后从左往右扫就好

假设当前扫到第c个数，那么你只需要知道这c个数中所有不同的数倒数第二次出现的位置就好了

比如对于数字x，它倒数第二次出现的位置是y，那么凡是访问区间右端点为c，左端点在y及以前的，

数字x都对该区间有贡献，否则没贡献，最后所有不同数字的贡献加在一起就是答案

那么该如何处理呢？

pri[i]表示位置i上的数字上一次出现的位置是pri[i]，初始化所有位置权值为0

每次扫到第c个数时，将位置pri[c]的值+1，位置pri[pri[c]]的值-1

假设查询区间是[L, c]，位置1到位置L的权值之和就是答案

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
typedef struct Res
{
	int id;
	int x, y;
	bool operator < (const Res &b) const
	{
		if(y<b.y)
			return 1;
		return 0;
	}
}Res;
Res s[1000005];
int n, a[1000005], pos[1000005], pre[1000005], ans[1000005], tre[1000005];
void Update(int k, int x)
{
	while(k)
	{
		tre[k] += x;
		k -= k&-k;
	}
}
int Query(int k)
{
	int sum = 0;
	while(k<=n)
	{
		sum += tre[k];
		k += k&-k;
	}
	return sum;
}
int main(void)
{
	int c, q, i, now;
	while(scanf("%d%d%d", &n, &c, &q)!=EOF)
	{
		memset(pos, 0, sizeof(pos));
		for(i=1;i<=n;i++)
		{
			scanf("%d", &a[i]);
			pre[i] = pos[a[i]];
			pos[a[i]] = i;
		}
		for(i=1;i<=q;i++)
		{
			scanf("%d%d", &s[i].x, &s[i].y);
			s[i].id = i;
		}
		sort(s+1, s+q+1);
		memset(tre, 0, sizeof(tre));
		now = 1;
		for(i=1;i<=n;i++)
		{
			Update(pre[i], 1);
			Update(pre[pre[i]], -1);
			while(s[now].y==i && now<=q)
				ans[s[now].id] = Query(s[now].x), now++;
		}
		for(i=1;i<=q;i++)
			printf("%d\n", ans[i]);
	}
	return 0;
}