【莫队入门题】SPOJ - DQUERY D-query

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;
}