题目大意
给出一个序列,现在有很多询问,每次询问一个区间内的不同种类数的个数。
解题思路
首先树状数组的题应该想到这两张图片:
假设给出的颜色序列为
1
,
1
,
1
,
3
,
2
,
4
,
3
1,1,1,3,2,4,3
1,1,1,3,2,4,3,手玩一下查询区间,能够发现一个十分重要的性质:当
r
r
r固定时,左边出现了重复的数,会产生相同的贡献,为了避免重复每次只考虑计算最靠右的数的贡献。那么需要记录每个数从左向右更新时记录下之前出现的位置,然后
u
p
d
a
t
e
(
p
r
e
_
p
o
s
,
−
1
)
update(pre\_pos,-1)
update(pre_pos,−1)消除之前的贡献
例如 1 , 1 , 1 , 3 1,1,1,3 1,1,1,3,假设当前部分查询的右界都为 4 4 4,那么一开始更新时,树状数组的大小变化为(联系上面的图,实际上后缀的变化是和二进制的每一位相关的,但是查询前缀和与这一点没有关系):
- 第一个数为 1 1 1,之前没有出现,那么树状数组为 1 , 0 , 0 , 0 1,0,0,0 1,0,0,0
- 第二个数为 1 1 1,之前记录出现的位置为 1 1 1,那么树状数组为 0 , 1 , 0 , 0 0,1,0,0 0,1,0,0
- 第三个数为 1 1 1,之前记录出现的位置为 2 2 2,那么树状数组为 0 , 0 , 1 , 0 0,0,1,0 0,0,1,0
- 第四个数为 3 3 3,之前没有出现,那么树状数组为 0 , 0 , 1 , 1 0,0,1,1 0,0,1,1
- 得到最新的树状数组 0 , 0 , 1 , 1 0,0,1,1 0,0,1,1后,查询 [ 1 , 4 ] , [ 2 , 4 ] , [ 3 , 4 ] , [ 4 , 4 ] [1,4],[2,4],[3,4],[4,4] [1,4],[2,4],[3,4],[4,4]这些区间时,显然都是正确的
因为查询的区间和每个数的位置有关,与数的大小无关,那么显然是按照对应的 i d id id建树。考虑到上面的性质,而且树状数组查询的是前缀和,那么先保存下所有的右端点,然后按升序排序,最后离线更新和查询。
实际上左端点以及没有了任何用处,排序时右端点相同也没必要对左端点进行升序/降序处理。我们只需要维护一个不断向右更新的指针,当然指针不能超过右端点排序后每个位置的右界,然后更新过一个右端点的前缀,当需要查询这个以这个右端点为右界的区间时,直接类似前缀和的查询 g e t S u m ( r ) − g e t S u m ( l − 1 ) getSum(r)-getSum(l-1) getSum(r)−getSum(l−1)
总之思想还是很巧妙的,讲的思路不算清晰,但也是目前理解 B I T BIT BIT的极限了
//
// Created by Happig on 2020/8/23
//
#include <bits/stdc++.h>
#include <unordered_map>
#include <unordered_set>
using namespace std;
#define fi first
#define se second
#define pb push_back
#define ins insert
#define Vector Point
#define lowbit(x) (x&(-x))
#define mkp(x, y) make_pair(x,y)
#define mem(a, x) memset(a,x,sizeof a);
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef pair<double, double> pdd;
const double eps = 1e-8;
const double pi = acos(-1.0);
const int inf = 0x3f3f3f3f;
const double dinf = 1e300;
const ll INF = 1e18;
const int Mod = 1e9 + 7;
const int maxn = 1e6 + 10;
struct node{
int l,r,id;
bool operator < (const node &p)const{
return r<p.r;
}
}p[maxn];
int a[maxn],t[maxn],pre[maxn],ans[maxn];
bool vis[maxn];
int n;
void update(int i,int k){
while(i<=n){
t[i]+=k;
i+=lowbit(i);
}
}
ll getSum(int i){
ll ans=0;
for(;i;i-=lowbit(i)){
ans+=t[i];
}
return ans;
}
int main() {
//freopen("in.txt", "r", stdin);
//freopen("out.txt", "w", stdout);
ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int m;
cin>>n;
for(int i=1;i<=n;i++) cin>>a[i];
cin>>m;
for(int i=1;i<=m;i++){
cin>>p[i].l>>p[i].r;
p[i].id=i;
}
sort(p+1,p+1+m);
int s=1;
for(int i=1;i<=m;i++){
while(s<=p[i].r){
if(pre[a[s]]) update(pre[a[s]],-1);
update(s,1);
pre[a[s]]=s;
s++;
}
ans[p[i].id]=getSum(p[i].r)-getSum(p[i].l-1);
}
for(int i=1;i<=m;i++) cout<<ans[i]<<"\n";
return 0;
}