D-query SPOJ - DQUERY(莫队算法)

这题的解法并不唯一,在网上看到的主要有主席树和莫队算法两种,但主席树的代码普遍比较长而且较难理解。而莫队算法代码量短而且叫容易理解。莫队算法也是一种暴力的算法,复杂度比纯暴力快了不少,但比主席树慢了一点。
D-query SPOJ - DQUERY(莫队算法)_第1张图片
D-query SPOJ - DQUERY(莫队算法)_第2张图片
题目大意:在数列a中,找出[L,R]中有多少个不同的数。

//#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define lowbit(x) ((x) & -(x))
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1
const int maxn = 1e6;
typedef long long ll;
using namespace std;
const ll mod = 1e9 + 7;
const double pi = acos(-1.0);
const double c = 0.57721566490153286060651209;
const double esp = 1e-8;
const int INF = 0x7fffffff;
inline int read() {
     
    int x = 0, f = 1;
    char ch = getchar();
    while (ch < '0' || ch > '9') {
     
        if (ch == '-')
            f = -1;
        ch = getchar();
    }
    while (ch >= '0' && ch <= '9') {
     
        x = (x << 1) + (x << 3) + (ch ^ 48);
        ch = getchar();
    }
    return x * f;
}
int n, q, sz, ans;
int a[maxn], block[maxn], vis[maxn], Ans[maxn];
struct node{
     
    int l, r, id;
}Q[maxn];
bool cmp(node a, node b){
     		// 莫队算法的核心之一,基本可以作为模板
    if (block[a.l] == block[b.l])
        return a.r < b.r;
    return a.l < b.l;
}
void add(int x){
     
    if (vis[a[x]] == 0)
        ans ++;
    vis[a[x]] ++;
}
void del(int x){
     
    vis[a[x]] --;
    if (vis[a[x]] == 0)
        ans --;
}
int main() {
     
    //ios::sync_with_stdio(false);
    //freopen("text.txt","r",stdin);
    //freopen("out1.txt","w",stdout);
    scanf("%d",&n);
    sz = sqrt(n);
    for (int i = 1; i <= n; i ++){
     
        scanf("%d",&a[i]);
        block[i] = (i - 1) / sz + 1;	// 分块操作
    }
    scanf("%d",&q);
    for (int i = 1; i <= q; i ++){
     
        scanf("%d %d",&Q[i].l,&Q[i].r);
        Q[i].id = i;		// 记录查询的第几次
    }
    sort(Q+1,Q+1+q,cmp);	// 对区间进行排序
    int l = 1, r = 0;
    for (int i = 1; i <= q; i ++){
        // 莫队算法的核心,先扩后缩。add函数和del函数根据实际情况改变,这也是莫队算法的难点。 
        while (l > Q[i].l)
            add(--l);
        while (r < Q[i].r)
            add(++r);
        while (l < Q[i].l)  
            del(l ++);
        while (r > Q[i].r)
            del(r --);
        Ans[Q[i].id] = ans;
    }
    for (int i = 1; i <= q; i ++)
        printf("%d\n",Ans[i]);
    return 0;
}

你可能感兴趣的:(D-query SPOJ - DQUERY(莫队算法))