题目链接;传送门
题目大意:有一个n个数的数列,并由q个询问,每一个询问有一个l和r,问你在区间a1—al和ar—an这两个区间中有多少个不同的数。
题目思路:
题目表面上意思是求两个区间中不同数的个数,因为数据范围比较大,因此需要使用树状数组,并且使用离线操作。
首先我们先考虑如何求一个区间中不同的数的个数,我们可以先对询问按照他的区间右端点的大小进行排序。
然后每当一个数出现,如果这个数出现过就需要对他之前的位置减1,然后在这个位置上加1.
区间(l,r)中不同数的个数就是(1,r)减去(1,l-1)。
因为对相同的数不断的向后更新,所以可以避免(1,l-1)和(l,r)中的重复数被误减。
接下来考虑,我们这道题是前后两段区间。如何合并成一个区间?
我们可以用倍增的思想,将a[n+i]=a[i]。这样的话右区间和左区间就在中间连接在一起了,形成了一个大的区间。
#include <bits/stdc++.h>
using namespace std;
const int maxn=100005*2;
int a[maxn];
int vis[maxn];
int sum[maxn];
int bit[maxn];
struct node
{
int l,r,id;
bool operator <(const node &w)const
{
return r<w.r;
}
}q[maxn];
int lowbit(int x)
{
return x&(-x);
}
void add(int x,int d)
{
while(x<maxn)
{
bit[x]+=d;
x+=lowbit(x);
}
}
int getsum(int x)
{
int res=0;
while(x)
{
res+=bit[x];
x-=lowbit(x);
}
return res;
}
int main()
{
int n,m;
while(~scanf("%d%d",&n,&m))
{
memset(bit,0,sizeof bit);
memset(vis,0,sizeof bit);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
a[i+n]=a[i];
}
for(int i=1;i<=m;i++)
{
scanf("%d",&q[i].r);
q[i].r+=n;
scanf("%d",&q[i].l);
q[i].id=i;
}
sort(q+1,q+m+1);
int cur=1;
for(int i=1;i<=2*n&&cur<=m;i++)
{
if(vis[a[i]])
{
add(vis[a[i]],-1);
}
vis[a[i]]=i;
add(vis[a[i]],1);
while(cur<=m&&q[cur].r<=i)
{
sum[q[cur].id]=getsum(q[cur].r)-getsum(q[cur].l-1);
cur++;
}
}
for(int i=1;i<=m;i++)
{
printf("%d\n",sum[i]);
}
}
return 0;
}
/*
3 2
1 2 1
1 2
1 3
4 1
1 2 3 4
1 3
2
1
3
*/