题目链接: 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;
}