P2709 小B的询问

Posted 2021-01-06 garen-wang

莫队第二道，有点感觉。。

只不过这种感觉建立在无修改上面

这道题跟上面一道HH的项链是一样的。只不过要你求的变成了平方和。

我就非常的naive，在莫队的时候没有用sum记录，暴力去统计，结果全部超时GG。

其实只需要减掉原来的平方和，再加上新的平方和就可以了。

其他的基本都差不多。

代码：

#include<cstdio>
#include<cmath>
#include<algorithm>
const int maxn = 50005;
struct Nodes
{
    int l, r, idx;
} s[maxn];
int a[maxn];
int belong[maxn];
int len, num;
int n, m, k;
int ans[maxn];
int cnt[maxn];
int read()
{
    int ans = 0, s = 1;
    char ch = getchar();
    while(ch > ‘9‘ || ch < ‘0‘){ if(ch == ‘-‘) s = -1; ch = getchar(); }
    while(ch >= ‘0‘ && ch <= ‘9‘) ans = (ans << 3) + (ans << 1) + ch - ‘0‘, ch = getchar();
    return s * ans;
}
bool cmp(const Nodes x, const Nodes y)
{
    if(belong[x.l] == belong[y.l]) return x.r < y.r;
    return belong[x.l] < belong[y.l];
}
void init()
{
    len = sqrt(n);
    for(int i = 1; i <= n; i++) belong[i] = (i - 1) / len + 1;
}
int extend(int l, int r, bool right, bool add)
{
    if(right)
    {
        if(add)
        {
            cnt[a[r]]++;
            return cnt[a[r]] * cnt[a[r]] - (cnt[a[r]] - 1) * (cnt[a[r]] - 1);
        }
        else
        {
            cnt[a[r]]--;
            return cnt[a[r]] * cnt[a[r]] - (cnt[a[r]] + 1) * (cnt[a[r]] + 1);
        }
    }
    else
    {
        if(add)
        {
            cnt[a[l]]++;
            return cnt[a[l]] * cnt[a[l]] - (cnt[a[l]] - 1) * (cnt[a[l]] - 1);
        }
        else
        {
            cnt[a[l]]--;
            return cnt[a[l]] * cnt[a[l]] - (cnt[a[l]] + 1) * (cnt[a[l]] + 1);
        }
    }
}
void solve()
{
    int l = 1, r = 0, sum = 0;
    for(int i = 1; i <= m; i++)
    {
        int ql = s[i].l, qr = s[i].r;
        while(r < qr) sum += extend(l, ++r, true, true);
        while(r > qr) sum += extend(l, r--, true, false);
        while(l > ql) sum += extend(--l, r, false, true);
        while(l < ql) sum += extend(l++, r, false, false);

        ans[s[i].idx] = sum;
    }
}
int main()
{
    n = read(), m = read(), k = read();
    for(int i = 1; i <= n; i++) a[i] = read();
    for(int i = 1; i <= m; i++) s[i].l = read(), s[i].r = read(), s[i].idx = i;
    init();
    std::sort(s + 1, s + m + 1, cmp);
    solve();
    for(int i = 1; i <= m; i++) printf("%d
", ans[i]);
    return 0;
}