1. 程式人生 > 其它 >[演算法]莫隊演算法

[演算法]莫隊演算法

技術標籤:資料結構優化

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; }