牛客网暑期ACM多校训练营（第一场）J. Different Integers 莫队算法 优化

题目链接: J. Different Integers

题目大意

一个数组, 长度n<=1e5, q次询问(l, r), 输出区间[1, l], [r, n]中不同数字的个数, q<=1e5

思路

莫队算法, 裸题, 需要一点优化

有一个重大的坑点就是l可能会大于r, 但是看到题面没有说l<r我就觉得有问题了
读取(l, r)的时候判断一下l和r的大小关系, 如果l>=r, 直接让r=l+1(if (querys[i].l >= querys[i].r) querys[i].r = querys[i].l + 1;), 否则会TLE

但只这么做还是会TLE

我将分块大小block_size由sqrt(n)改成固定的sqrt(maxn)就从TLE变成AC了

还有一种优化方法是, 将查询(l, r)分块, 如果l块号不同, 按块号排序, 如果块号相同, 则第一块中按r从小到大排序, 第二块中按从大到小排序, 第三块中按从小到大排序……会快一些

代码

语言：C++14 代码长度：3320 运行时间： 1537 ms 占用内存：8408K 运行结果：答案正确

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;
const int maxn = 1e5 + 100;
typedef long long ll;
int n, a[maxn];
int q;
int block[maxn];
int cnt[maxn];
int cnt_dif;

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*10+ch-'0';ch=getchar();}
    return x*f;
}

struct Query
{
    int l, r, id;
    int ans;
} querys[maxn];

bool cmp(const Query & x, const Query & y)
{
    if (block[x.l] == block[y.l]) return x.r < y.r;
    return x.l < y.l;
}
bool cmp_id(const Query & x, const Query & y)
{
    return x.id < y.id;
}

void update(int p, int add)
{
    if (add == -1)
    {
        if (cnt[a[p]] == 1) --cnt_dif;
        cnt[a[p]]--;
    }
    else
    {
        if (cnt[a[p]] == 0) ++cnt_dif;
        cnt[a[p]]++;
    }
}


int main()
{
    while (scanf("%d%d", &n, &q) == 2)
    {
        memset(cnt, 0, sizeof(int) * (1 + n));
        cnt_dif = 0;
        for (int i = 1; i <= n; ++i)
        {
            a[i] = read();
            if (cnt[a[i]] == 0) ++cnt_dif;
            cnt[a[i]]++;
        }
        int block_size = (int)sqrt(maxn);
        for (int i = 1; i <= n; ++i)
            block[i] = (i - 1) / block_size + 1;
        for (int i = 1; i <= q; ++i)
        {
            querys[i].l = read();
            querys[i].r = read();
            if (querys[i].l >= querys[i].r) querys[i].r = querys[i].l + 1;
            querys[i].id = i;
        }

        sort(querys + 1, querys + 1 + q, cmp);

        int l = 1, r = 2;
        for (int i = 1; i <= q; ++i)
        {
            for (; r < querys[i].r; ++r) update(r, -1);
            for (; r > querys[i].r; --r) update(r - 1, 1);
            for (; l < querys[i].l; ++l) update(l + 1, 1);
            for (; l > querys[i].l; --l) update(l, -1);
            querys[i].ans = cnt_dif;
        }
        sort(querys + 1, querys + 1 + q, cmp_id);
        for (int i = 1; i <= q; ++i) printf("%d\n", querys[i].ans);
    }

    return 0;
}