[演算法]莫隊演算法
阿新 • • 發佈:2020-12-31
2019.04.19:莫隊演算法主要的降低時間複雜度的技巧是分塊,可以把詢問離線然後分塊做,塊的大小(\sqrt n),即可。同塊右端點排序,不同塊左端點排序。不同題目分塊之後處理的方法不同,具體題目具體分析。
https://www.luogu.org/problemnew/show/P2709
#include<bits/stdc++.h>
#define maxn 50005
using namespace std;
int a[maxn], cnt[maxn];
long long ans[maxn];
struct ask{
int l, r, id;
} q[maxn];
bool cmp(const ask &a, const ask &b){return a.l / 500 == b.l / 500 ? a.r < b.r : a.l < b.l;}
int main()
{
int n, m, k;
scanf("%d %d %d", &n, &m, &k);
for (int i = 1; i <= n; ++i) scanf("%d", &a[i]);
for (int i = 1; i <= m; ++ i) scanf("%d %d", &q[i].l, &q[i].r), q[i].id = i;
sort(q, q + m, cmp);
int l = 1, r = 0;
long long sum = 0;
for (int i = 1; i <= m; ++i){
while (l > q[i].l) --l, ++cnt[a[l]], sum += 2 * cnt[a[l]] - 1;
while (r < q[i].r) ++r, ++cnt[a[r]], sum + = 2 * cnt[a[r]] - 1;
while (l < q[i].l) sum += 1 - 2 * cnt[a[l]], --cnt[a[l]], ++l;
while (r > q[i].r) sum += 1 - 2 * cnt[a[r]], --cnt[a[r]], --r;
ans[q[i].id] = sum;
}
for (int i = 1; i <= m; ++i) printf("%lld\n", ans[i]);
return 0;
}