题目传送门:https://www.luogu.org/problemnew/show/P1972
题意:
有n个数,m个询问,每个询问有x、y,求x~y区间中不同数的个数。
思路1:
树状数组+离线。
用last[i]表示上一个数为i个数的编号,每一次用树状数组将last这个位置的值改为0(相当于减1),然后在当前这个点的值改为1(相当于加1)。为什么呢?
如上图,我们在统计区间2时,可以发现区间2与第一个点无关,可以将其忽略掉。
设当前访问到的点的编号为p,则如果上一个颜色与点p相同的不再这一段区间内,则对答案的贡献就为0。
所以我们先排一边序(第一关键字为右编号,第二关键字为左编号),因为我们从左往右扫,处理区间时优先处理靠左的区间。
代码:
#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[200010];
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]);
}
思路2:
莫队模板(详见我的blog)。
代码:
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
int n,m,now;
int p[500010],block[500010],hsh[500010],ans[500010];
struct node{int x,y,id;} a[200010];
bool cmp(node x,node y)
{
return block[x.x]==block[y.x]?x.y<y.y:x.x<y.x;
}
void init()
{
int u=sqrt(n);
for(int i=1;i<=n;i++)
block[i]=(i-1)/u+1;
}
void move(int x,int d)
{
if(d)
{
if(!hsh[p[x]]) now++;
hsh[p[x]]++;
}
else
{
hsh[p[x]]--;
if(!hsh[p[x]]) now--;
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&p[i]);
scanf("%d",&m);
init();
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 l=0,r=0;
for(int i=1;i<=m;i++)
{
while(l<a[i].x) move(l++,0);
while(l>a[i].x) move(--l,1);
while(r<a[i].y) move(++r,1);
while(r>a[i].y) move(r--,0);
ans[a[i].id]=now;
}
for(int i=1;i<=m;i++)
printf("%d\n",ans[i]);
}