首先来看一道例题:
Description
有n个数字,以及m个查询。每次查询的格式是L,r,求L~r(左右包含)这个区间内有多少个不同的数?
1< n,m<=50000,1<=L<r<=n,数列中元素大小<= \(10^9\) 。
思路1:
自然而然想到暴力,对每次询问L~r枚举区间,考虑数列中元素取值范围较大,先要进行离散化处理。
时间复杂度:O(nm)
代码略
路2:
线段树或树状数组处理一下。(代码复杂度相对而言较高)。
#include<cstdio>
#include<algorithm>
#define lowbit(x) ((x)&(-(x)))
using namespace std;
int n,m;
int b[500010],c[500010],last[1000010],ans[500010];
struct node{int x,y,id;} a[500010];
bool cmp(node x,node y)
{
return x.y==y.y?x.x<y.x:x.y<y.y;
}
void add(int x,int y)
{
while(x<=n)
{
c[x]+=y;
x+=lowbit(x);
}
}
int getsum(int x)
{
int sum=0;
while(x)
{
sum+=c[x];
x-=lowbit(x);
}
return sum;
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&b[i]);
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
scanf("%d %d",&a[i].x,&a[i].y);
a[i].id=i;
}
sort(a+1,a+m+1,cmp);
int now=1;
for(int i=1;i<=n;i++)
{
if(last[b[i]]) add(last[b[i]],-1);
last[b[i]]=i;
add(i,1);
while(i==a[now].y&&now<=m)
{
ans[a[now].id]=getsum(a[now].y)-getsum(a[now].x-1);
now++;
}
if(now==m+1) break;
}
for(int i=1;i<=m;i++)
printf("%d\n",ans[i]);
return 0;
}