题目大意:
给出一个 n 表示有 n 个数,然后接下来输入n个数,紧接着输入一个 m 表示有 m 组询问,然后有 m 行,每行输入l r ,要求输出区间[l, r]不同数字的种数有多少。
解题思路:
莫队算法,离线扫描即可,将n个数分块,每块 根号n ,然后对询问的区间进行排序,左端点升序排序,相同时再对右端点基于奇偶排序(玄学优化),之后设左指针和右指针和计数数组cnt,根据左右指针的移动来判断种类的加减,注意运算符和优先级(这里是由add 和 del 函数压缩过来的,运算符错一点就会 WA)。
Code:
#pragma GCC optimize(2)
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <cstring>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
const int N = 1e6 +50;
int a[N], cnt[N], belong[N], ans[N], now;
struct node { int l, r, id; }q[N];
bool cmp(node a, node b)
{
return belong[a.l] ^ belong[b.l] ? belong[a.l] < belong[b.l] : ((belong[a.l] & 1) ? a.r < b.r : a.r > b.r);
}
int main()
{
ios::sync_with_stdio(false);
int n, m;
cin >> n;
double size = sqrt(n);
int num = ceil(n / size);
for (int i = 1; i <= num; i ++)
for (int j = (i - 1) * size + 1; j <= i * size; j ++)
belong[j] = i;
for (int i = 1; i <= n; i ++)
cin >> a[i];
cin >> m;
for (int i = 1; i <= m; i ++)
{
cin >> q[i].l >> q[i].r;
q[i].id = i;
}
sort(q + 1, q + 1 + m, cmp);
int l = 1, r = 0;
for (int i = 1; i <= m; i ++)
{
int ql = q[i].l, qr = q[i].r;
while (l < ql) now -= !-- cnt[a[l ++]];
while (l > ql) now += !cnt[a[--l]] ++;
while (r < qr) now += !cnt[a[++ r]] ++;
while (r > qr) now -= !-- cnt[a[r --]];
ans[q[i].id] = now;
}
for (int i = 1; i <= m; i ++)
cout << ans[i] << endl;
return 0;
}