当时学的时候一脸懵啊——什么诡异的算法?
给你一个区间,让你找区间最值。(这不是线段树/树状数组吗 ?)
普通算法是一个个查找,时间复杂度是O(n*q),如果问题多了就凉凉了。
而莫队则是在已查找基础上向两边拓展。你也许会说,如果询问差距很大,比如这样:
2 5
100 130
反而得不偿失。
别急,下面才是莫队的精髓:
小B的询问
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
struct node {
int l, r, in;
}s[50002];
int n, m, k, a[50002], fa, u, v, cnt[50002];
long long ans[50002], res;
bool cmp(const node a, const node b) {
return (a.l / fa == b.l / fa) ? a.r < b.r : a.l < b.l;
}
void add(const int x) {
if(a[x] > k)
return;
res += ((cnt[a[x]]) << 1) + 1;
cnt[a[x]] ++;
}
void remove(const int x) {
if(a[x] > k)
return;
res -= ((-- cnt[a[x]]) << 1) + 1;
}
int main() {
scanf("%d %d %d", &n, &m, &k);
fa = sqrt(n);
for(int i = 1; i <= n; i ++)
scanf("%d", &a[i]);
for(int i = 1; i <= m; i ++) {
scanf("%d %d", &s[i].l, &s[i].r);
s[i].in = i;
}
sort(s + 1, s + m + 1, cmp);
u = 1;
for(int i = 1; i <= m; i ++) {
while(u < s[i].l)
remove(u ++);
while(u > s[i].l)
add(-- u);
while(v < s[i].r)
add(++ v);
while(v > s[i].r)
remove(v --);
ans[s[i].in] = res;
}
for(int i = 1; i <= m; i ++)
printf("%lld\n", ans[i]);
return 0;
}
玄学分块!接下来证明其时间复杂度为sqrt(n)*n
- 左端点 在同一块内最多移动sqrt(n)*块内个数,则有O(sqrt(n)m)。(看了好多博客都没有m,不太懂)
- 右端点 对于每一个块右端点都是有序的,则一个块最多n,总共为O(sqrt(n)n)。
注意一下add与remove传值的不同哦~
如果有不对的地方,请大佬扶贫。