HH有一串由各种漂亮的贝壳组成的项链。HH相信不同的贝壳会带来好运,所以每次散步 完后,他都会随意取出一
段贝壳,思考它们所表达的含义。HH不断地收集新的贝壳,因此他的项链变得越来越长。有一天,他突然提出了一
个问题:某一段贝壳中,包含了多少种不同的贝壳?这个问题很难回答。。。因为项链实在是太长了。于是,他只
好求助睿智的你,来解决这个问题。
第一行:一个整数N,表示项链的长度。
第二行:N个整数,表示依次表示项链中贝壳的编号(编号为0到1000000之间的整数)。
第三行:一个整数M,表示HH询问的个数。
接下来M行:每行两个整数,L和R(1 ≤ L ≤ R ≤ N),表示询问的区间。
N ≤ 50000,M ≤ 200000。
M行,每行一个整数,依次表示询问对应的答案。
6 1 2 3 4 3 5 3 1 2 3 5 2 6
2 2 4
题意:
有一串项链,每点有个颜色,给你个区间。查询这个区间内有几种不同的颜色?
思路:
预处理出上一个与i颜色相同的点pre[i],将询问按右端点排序,然后一边往树状数组进入点一边处理询,注意其中一些小技巧。
代码:
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
int n,m,sum;
const int maxn=1000010;
int pre[maxn],s[maxn],p[maxn],head[maxn],ans[maxn];
struct Q
{
int l,r,id;
}q[maxn];
bool cmp(Q a,Q b)
{
return a.r<b.r;
}
int lowbit(int i){
return i&-i;
}
void updata(int x,int v) //单点更新
{
if(!x)
return ;
sum+=v; //记录的更新的次数
// cout<<sum<<"......"<<endl;
while(x<=n){
s[x]+=v;
x+=lowbit(x);
}
}
int query(int x) //查询0-x和
{
int i,res=0;
while(x>0){
res+=s[x];
x-=lowbit(x);
}
return res;
}
int main()
{
scanf("%d",&n);
int i,j;
for(i=1;i<=n;i++)
{
scanf("%d",&p[i]);
pre[i]=head[p[i]]; //仔细理解
head[p[i]]=i;
}
scanf("%d",&m);
for(i=1;i<=m;i++) scanf("%d%d",&q[i].l,&q[i].r),q[i].id=i;
sort(q+1,q+m+1,cmp);
for(i=j=1;i<=m;i++)
{
for(;j<=q[i].r;j++) updata(pre[j],-1),updata(j,1); //核心处
ans[q[i].id]=sum-query(q[i].l-1);
}
for(i=1;i<=m;i++) printf("%d\n",ans[i]);
return 0;
}
多看 多学 多想