D-query SPOJ - DQUERY（莫队算法）

最新推荐文章于 2021-03-13 15:20:42 发布

zzuliwyl

最新推荐文章于 2021-03-13 15:20:42 发布

阅读量190

点赞数

本文链接：https://blog.csdn.net/HPYGQ/article/details/108410899

版权

这题的解法并不唯一，在网上看到的主要有主席树和莫队算法两种，但主席树的代码普遍比较长而且较难理解。而莫队算法代码量短而且叫容易理解。莫队算法也是一种暴力的算法，复杂度比纯暴力快了不少，但比主席树慢了一点。
在这里插入图片描述

题目大意：在数列a中，找出[L,R]中有多少个不同的数。

//#include <bits/stdc++.h>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <deque>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <stack>
#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;
}