题目大意:
给出含N个数的一段序列,M个询问,每次询问一个区间[l,r],回答[l,r]中不同的数有多少个。
数为编号为0到1000000之间的整数
1 ≤ L ≤ R ≤ N
N ≤ 50000,M ≤ 200000
题解:
以前对树状数组只是有些了解,
今天才终于弄明白了实质,
不是只记得x & (-x)的我了..
这题如果在线做,O(MN^2),显然吃不消..
我们考虑一下离线,
对于r我们升序排列,然后我们就可以用前缀和乱搞了
对于一个i位置的数j
我们可以将上一个j出现的位置k
[k+1,n]的位置全部+1,[i+1,n]全部-1
这样的话就可以保证每次查询不会有错漏
因为这样前缀和时间为O(NM)依然会超时…
所以我们加个树状数组优化一下就可以了
时间复杂度:(M log N)
代码:
#include<bits/stdc++.h>
#define N 200005
using namespace std ;
struct node{
int l,r,cnt;
}a[N];
int next[N],ans[N],c[N],ls[1000005],n,m,rp;
int lowbit(int x){
return x & (-x);
}
void insert(int x ,int y){
if (x == n) return;
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 cmp(node aa, node bb){
return aa.r < bb.r;
}
int main(){
scanf("%d",&n);
int s;
for (int i=1; i<=n; i++){
scanf("%d",&s);
next[i]=ls[s];
ls[s]=i;
}
scanf("%d",&m) ;
for (int i=1; i<=m; i++){
scanf("%d%d",&a[i].l,&a[i].r);
a[i].cnt=i;
}
sort(a+1,a+1+m,cmp);
for (int i=1; i<=m; i++){
while (rp < a[i].r){
rp++;
insert(next[rp]+1,1);
insert(rp+1,-1);
}
ans[a[i].cnt]=getsum(a[i].l);
}
for (int i=1; i<=m; i++) printf("%d\n",ans[i]);
return 0;
}