HH的项链
题目链接
题目描述
HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运,所以每次散步完后,他都会随意取出一段贝壳,思考它们所表达的含义。HH 不断地收集新的贝壳,因此,他的项链变得越来越长。
有一天,他突然提出了一个问题:某一段贝壳中,包含了多少种不同的贝壳?这个问题很难回答…… 因为项链实在是太长了。于是,他只好求助睿智的你,来解决这个问题。
输入格式
一行一个正整数
n
n
n,表示项链长度。
第二行
n
n
n个正整数
a
i
a_i
ai,表示项链中第
i
i
i个贝壳的种类。
第三行一个整数
m
m
m,表示 HH 询问的个数。
接下来
m
m
m行,每行两个整数
l
l
l,
r
r
r表示询问的区间。
输出格式
输出 m m m行,每行一个整数,依次表示询问对应的答案。
输入输出样例
输入
6
1 2 3 4 3 5
3
1 2
3 5
2 6
输出
2
2
4
说明/提示
【数据范围】
对于20%的数据,
1
≤
n
,
m
≤
5000
1≤n,m≤5000
1≤n,m≤5000;
对于40%的数据,
1
≤
n
,
m
≤
1
0
5
1≤n,m≤10^5
1≤n,m≤105;
对于60%的数据,
1
≤
n
,
m
≤
5
∗
1
0
5
1≤n,m≤5*10^5
1≤n,m≤5∗105;
对于100%的数据,
1
≤
n
,
m
,
a
i
1
0
6
1≤n,m,a_i 10^6
1≤n,m,ai106,
1
≤
l
≤
r
≤
n
1≤l≤r≤n
1≤l≤r≤n。
本题可能需要较快的读入方式,最大数据点读入数据约
20
M
B
20MB
20MB。
思路
一开始是把它想成像SSL_2645一样,每一类贝壳是一个“盒子”,最后统计某一段一共有多少个“盒子”。
然后……
5个WA+8个TLE;
在这之后的很长一段时间里,本人百思不得其解,最后,不得不去向大佬们学习了一下。
正确的思路应该是:
- 由于相同的数字只被统计1次,所以我只需要标记某个数最后出现的位置就可以了;
- 由于询问是无序的,可能会出现询问的区间里明明有某个数,但由于这不是它最后一次出现,因此它没有被统计的情况,所以我要按照每个询问右端点的大小,将每个询问排序;
- 对于我当前维护的这一段序列,如果某个数在前边出现过,那么将它前边的位置标记为0,当前的位置标记为1,同时更新整颗线段树;
- 最后再统计该询问左右两端点之间1的个数即可;
- 实际上本题不需要快读,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;
}