[洛谷P2709]小B的询问

题目大意：有一个序列$s_i$（最大的数为$k$），有$m$个询问，询问$[l,r]$中$\sum_{i=1}^{k}c_i^2$（$c_i$表示数字$i$在$[l,r]$中的出现次数）。

题解：莫队

卡点：为什么我奇偶性优化会锅？？？（后记：发现了，用异或时，若($a.l==b.l\&\&a.r==b.r$时可能会出现$a<b=true\&\&b<a=true$）

C++ Code：

 

#include <cstdio>
#include <algorithm>
#define bsz 224
#define belong(i) ((i / bsz) + (i % bsz && 1))
#define maxn 50010
int n, m, k;
int ans[maxn];
int s[maxn], cnt[maxn];
struct node {
    int l, r, num;
    inline bool operator < (const node &rhs) const {return (belong(l) == belong(rhs.l)) ? ((belong(l) & 1) ? r < rhs.r : r > rhs.r) : l < rhs.l;}
} q[maxn];
int main() {
    scanf("%d%d%d", &n, &m, &k);
    for (int i = 1; i <= n; i++) scanf("%d", &s[i]);
    for (int i = 1; i <= m; i++) scanf("%d%d", &q[i].l, &q[i].r), q[i].num = i;
    std::sort(q + 1, q + m + 1);
    int l = 1, r = 1, res = 1; cnt[s[1]]++;
    for (int i = 1; i <= m; i++) {
        while (l > q[i].l) res += cnt[s[--l]]++ << 1 | 1;
        while (r < q[i].r) res += cnt[s[++r]]++ << 1 | 1;
        while (l < q[i].l) res -= --cnt[s[l++]] << 1 | 1;
        while (r > q[i].r) res -= --cnt[s[r--]] << 1 | 1;
        ans[q[i].num] = res;
    }
    for (int i = 1; i <= m; i++) printf("%d\n", ans[i]);
    return 0;
}

 

　　

 

转载于:https://www.cnblogs.com/Memory-of-winter/p/9533812.html