luogu 2709 小B的询问 莫队

Posted kkkkahlua

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了luogu 2709 小B的询问 莫队相关的知识,希望对你有一定的参考价值。

题目链接

Description

小B有一个序列,包含\(N\)\(1-K\)之间的整数。他一共有\(M\)个询问,每个询问给定一个区间\([L..R]\),求\(\sum_{i=1}^{K}c_i^2\),其中\(c_i\)表示数字\(i\)\([L..R]\)中的重复次数。小B请你帮助他回答询问。

思路

裸的莫队。

\(cnt[x]\)记录\(x\)的出现次数,增量为\((cnt[x]+1)^2-cnt[x]^2=2*cnt[x]+1\)\(-2*cnt[x]+1\).

Code

#include <bits/stdc++.h>
#define F(i, a, b) for (int i = (a); i < (b); ++i)
#define F2(i, a, b) for (int i = (a); i <= (b); ++i)
#define dF(i, a, b) for (int i = (a); i > (b); --i)
#define dF2(i, a, b) for (int i = (a); i >= (b); --i)
#define maxn 50010
using namespace std;
typedef long long LL;
struct node { int l, r,id; }a[maxn];
LL temp;
int n,m,k,blo,bl[maxn],v[maxn];
LL cnt[maxn],ans[maxn];
inline bool cmp(node&u, node& v) { return bl[u.l]<bl[v.l] || (bl[u.l]==bl[v.l]&&u.r<v.r); }
inline void upd(int x, int delta) { temp += delta*(cnt[x]<<1)+1; cnt[x]+=delta; }
int main() {
    scanf("%d%d%d", &n,&m,&k); blo = sqrt(n);
    F(i, 0, n) scanf("%d", &v[i]), bl[i]=i/blo;
    F(i, 0, m) {
        scanf("%d%d", &a[i].l, &a[i].r);
        --a[i].l, --a[i].r; a[i].id = i;
    }
    sort(a,a+m,cmp);
    int l=0, r=-1;
    temp=0;
    F(i, 0, m) {
        while (r<a[i].r) upd(v[++r],1);
        while (r>a[i].r) upd(v[r--],-1);
        while (l>a[i].l) upd(v[--l],1);
        while (l<a[i].l) upd(v[l++],-1);
        ans[a[i].id] = temp;
    }
    F(i, 0, m) printf("%lld\n", ans[i]);
    return 0;
}

以上是关于luogu 2709 小B的询问 莫队的主要内容,如果未能解决你的问题,请参考以下文章

[luogu2709] 小B的询问

[luoguP2709] 小B的询问(莫队)

P2709 小B的询问 莫队

luoguP2709 小B的询问 [莫队]

P2709 小B的询问 普通莫队

P2709 小B的询问 莫队算法