Luogu_P1972 HH的项链

HH的项链

题目链接

Luogu_P1972 HH的项链

题目描述

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

输入格式

一行一个正整数$n$，表示项链长度。
第二行$n$个正整数$a_i$，表示项链中第$i$个贝壳的种类。
第三行一个整数$m$，表示 HH 询问的个数。
接下来$m$行，每行两个整数$l$,$r$表示询问的区间。

输出格式

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

输入输出样例

输入

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

输出

2
2
4

说明/提示

【数据范围】

对于20%的数据，$1\le n,m\le 5000$；
对于40%的数据，$1\le n,m\le 10^5$；
对于60%的数据，$1\le n,m\le 5\ast 10^5$；
对于100%的数据，$1\le n,m,a_{i}10^6$，$1\le l\le r\le n$。
本题可能需要较快的读入方式，最大数据点读入数据约$20MB$。

思路

一开始是把它想成像SSL_2645一样，每一类贝壳是一个“盒子”，最后统计某一段一共有多少个“盒子”。
然后……

5个WA+8个TLE；

在这之后的很长一段时间里，本人百思不得其解，最后，不得不去向大佬们学习了一下。

正确的思路应该是：

1. 由于相同的数字只被统计1次，所以我只需要标记某个数最后出现的位置就可以了；
2. 由于询问是无序的，可能会出现询问的区间里明明有某个数，但由于这不是它最后一次出现，因此它没有被统计的情况，所以我要按照每个询问右端点的大小，将每个询问排序；
3. 对于我当前维护的这一段序列，如果某个数在前边出现过，那么将它前边的位置标记为0，当前的位置标记为1，同时更新整颗线段树；
4. 最后再统计该询问左右两端点之间1的个数即可；
5. 实际上本题不需要快读，scanf读入就够了；

代码

#include<algorithm>
#include<cstdio>
using namespace std;
struct str{
	int l,r,v,id;
}tre[4000001],que[1000001];
int n,m,s,st,en;
int a[1000001],pos[1000001],ans[1000001];
void bui(int k,int a,int b)
{
	tre[k].l=a;
	tre[k].r=b;
	if(a==b)
		return;
	int mid=(a+b)/2;
	bui(k*2,a,mid);
	bui(k*2+1,mid+1,b);
}
void add(int k,int i,int num)
{
	if(tre[k].l==tre[k].r)
		tre[k].v=num;
	else
	{
		if(i<=tre[k*2].r)
			add(k*2,i,num);
		else
			add(k*2+1,i,num);
		tre[k].v=tre[k*2].v+tre[k*2+1].v;
	}
}
int sum(int k,int a,int b)
{
	if(tre[k].l==a&&tre[k].r==b)
		return tre[k].v;
	else
	{
		if(b<=tre[k*2].r)
			return sum(k*2,a,b);
		else if(tre[k*2+1].l<=a)
			return sum(k*2+1,a,b);
		else
			return sum(k*2,a,tre[k*2].r)+sum(k*2+1,tre[k*2+1].l,b);
	}
}
bool cmp(str a,str b)
{
	return a.r<b.r;
}
int main()
{
	scanf("%d",&n);
	bui(1,1,n);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	scanf("%d",&m);
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&que[i].l,&que[i].r);
		que[i].id=i;
	}
	sort(que+1,que+m+1,cmp);//将每个询问按照右端点的由小到大 
	st=1;//本次维护的区间的起点
	for(int i=1;i<=m;i++)
	{
		en=que[i].r;//本次维护的区间的终点
		for(int j=st;j<=en;j++)
		{
			if(pos[a[j]])//如果数字a[j]在前边出现过 
				add(1,pos[a[j]],0);//将a[j]上次出现的位置标记为0
			add(1,j,1);//将a[j]本次出现的位置（即j）标记为1 
			pos[a[j]]=j;//记录下a[j]本次出现的位置
		}
		ans[que[i].id]=sum(1,que[i].l,que[i].r);
		st=que[i].r+1;//下次维护的区间的起点 
	}
	for(int i=1;i<=m;i++)
		printf("%d\n",ans[i]);
	return 0;
}