Step1 Problem:
给你 n 个数,q 个询问.
每个询问区间 L, R 不同数的个数。
Step2 Ideas:
如果知道区间 [L, R] 的结果,能 O(1) 求出区间 [L-1, R] 或 [L+1, R] 或 [L, R-1] 或 [L, R+1] 的结果,这时候可以使用莫队算法。
解决问题:我们可以把询问区间看成二维坐标系上的点,你如何画出最短距离的一笔经过的所有的点。
莫对核心思想:将长度为 n 的区间分成 根号n 块,每块区间大小为 根号n,对询问区间排序,按所在 块的编号 从小到大排序,如果所在 块的编号 一样,按 R 从小到大排序。
时间复杂度:
1:如果所有询问都在同一块:最坏情况,L 移动 n*sqrt(n),R 移动 n.
2:如果只有 sqrt(n) 个询问在同一块:最坏情况,L 移动 sqrt(n)*sqrt(n), R 移动 n.
3:如果只有 sqrt(n) 个询问在同一块,有 sqrt(n) 个块:sqrt(n) * ( sqrt(n)*sqrt(n) + n )
3:从一块移动到另一块:最坏情况,L 移动 sqrt(n),R 移动 n.
4:从一块移动到另一块最多有 sqrt(n) 次:sqrt(n) * ( sqrt(n) + n );
最坏情况:3 和 4 达到最坏情况:sqrt(n) * ( sqrt(n)sqrt(n) + n ) + sqrt(n)( sqrt(n) + n ) = n + 3*n*sqrt(n);
时间复杂度去掉常数:O(n*sqrt(n))
本题思路:
维护区间元素值出现的次数,如果从 0 变成 1 不同数的个数 +1,如果从 1 变成 0 不同数的个数 -1.
Step3 Code:
#include<bits/stdc++.h>
using namespace std;
const int N = 3e4+5;
const int M = 2e5+5;
const int maxa = 1e6+5;
struct node
{
int l, r, id;//询问区间,还有 l 所在哪个块
}Q[M];
int pos[N], a[N];
int ans[M];
bool cmp(node a, node b)//莫对算法的核心
{
if(pos[a.l] == pos[b.l]) return a.r < b.r;
return pos[a.l] < pos[b.l];
}
int L, R, Ans;
int vis[maxa];
void add(int x)//O(1) 更新区间
{
if(vis[a[x]] == 0) Ans++;
vis[a[x]]++;
}
void del(int x)//O(1) 更新区间
{
if(vis[a[x]] == 1) Ans--;
vis[a[x]]--;
}
int main()
{
int n, m;
scanf("%d", &n);
int block = sqrt(n);//莫对算法的核心
for(int i = 1; i <= n; i++)
{
scanf("%d", &a[i]);
pos[i] = i/block;//莫对算法的核心
}
scanf("%d", &m);
for(int i = 1; i <= m; i++)
{
scanf("%d %d", &Q[i].l, &Q[i].r);
Q[i].id = i;
}
sort(Q+1, Q+m+1, cmp);
L = 1, R = 0, Ans = 0;// L 和 R 初始化细节
memset(vis, 0, sizeof(vis));
for(int i = 1; i <= m; i++)
{
while(L < Q[i].l)
del(L), L++;
while(L > Q[i].l)
add(L-1), L--;
while(R < Q[i].r)
add(R+1), R++;
while(R > Q[i].r)
del(R), R--;
ans[Q[i].id] = Ans;
}
for(int i = 1; i <= m; i++)
printf("%d\n", ans[i]);
return 0;
}