给出一个长度为n 的数列,\(a_{1},a_{2},...,a_{n}\),有q 个询问,每个询问给出数对 \((i,j)\),需要你给出 \(a_{i},a_{i+1} ,...,a_j\) 这一段中有多少不同的数字
考虑到树状数组对于处理前缀和问题很方便,自然,尝试用树状数组来进行处理。设 \(tree[i]\) 表示前 \(i\) 个数不同的数的种类数。显然,这样处理是不够的。在计算的过程中,我们也要对 \(tree[i]\) 进行更新。
为了更方便地进行 \(tree[i]\) 的更新,我们将每个询问先保存下来,进行离线操作,按照 \(j\) 的由小到大的顺序进行排序。对于每个区间,尽可能统计其靠近右端点的不同的数的个数,用一个 \(last[i]\) 表示在输入序列下标为 \(i\) 的元素左边第一个值与它相同的元素的下标。对于一个数,它最后出现的位置越靠近当前枚举的右端点越好。以 \(k\) 为枚举的下标,如果 \(a[k]\) 之前有相同的数,那么减掉之前那个数的贡献,加上当前数的贡献。这样就能完成对 \(tree\) 数组的维护。
#include<bits/stdc++.h> using namespace std; const int MAXN = 1e6 + 5; int n,a[MAXN],tree[MAXN],rec[MAXN],last[MAXN],m,ans[MAXN]; struct askk{ int l,r,pos; }ask[MAXN]; bool cmp(askk aa,askk b){ return aa.r < b.r; } int lowbit(int i){ return i & -i; } void add(int x,int y){ while(x <= n){ tree[x] += y; x += lowbit(x); } } int sum(int x){ int ans = 0; while(x > 0){ ans += tree[x]; x -= lowbit(x); } return ans; } int main(){ ios::sync_with_stdio(false); cin.tie(0); cout.tie(0); cin >> n; for(int i = 1; i <= n; i++){ cin >> a[i]; last[i] = rec[a[i]]?rec[a[i]]:MAXN; rec[a[i]] = i;//维护 last 数组 } cin >> m; for(int i = 1; i <= m; i++){ cin >> ask[i].l >> ask[i].r; ask[i].pos = i; } int k = 1; sort(ask + 1,ask +1 + m,cmp);//对询问排序 for(int i = 1; i <= m; i++){ while(k <= ask[i].r){ add(last[k],-1); add(k,1);//维护tree数组 k++; } ans[ask[i].pos] = sum(ask[i].r) - sum(ask[i].l - 1); } for(int i = 1; i <= m; i++){ cout << ans[i] << "\n"; } }