NBUT - 1457 Sona 莫队

NBUT - 1457 Sona

题意：

计算一个区间内每种数出现次数的立方和，那么转移的话，假设一个区间是[L,R]，如果向左转移，用ans保存区间的答案，cnt[i]记录第 i 种数出现的次数，那么假设向左转移新加进区间的数字是a，由于a新加进区间，(cnt[a]+1)^3 = cnt[a]^3+3*cnt[a]^2+3*cnt[a]+1，cnt[a]++。数字移除区间反之。

因为颜色有1e9种（远大于n）所以需要离散化

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<sstream>
#include<cstring>
#include<bitset>
#include<cstdio>
#include<string>
#include<deque>
#include<stack>
#include<cmath>
#include<queue>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
const int maxn = 1e5+10;
struct node
{
    int l,r,id;
    ll sum;
}t[maxn];
int a[maxn],cnt[maxn],num[maxn],block,b[maxn];
ll ans;
bool cmp(node a,node b)
{
    if(num[a.l]==num[b.l])
        return a.r<b.r;
    return a.l<b.l;
}
bool cmp2(node a,node b)
{
    return a.id<b.id;
}
void add(int x)
{
    ans += 3*cnt[x]*(cnt[x]+1)+1;
    cnt[x]++;
}
void del(int x)
{
    ans += 3*cnt[x]*(1-cnt[x])-1;
    cnt[x]--;
}
int main()
{
    int l,r;
    int n,m;
    while(~scanf("%d",&n))
    {
        memset(cnt,0,sizeof(cnt));
        block = sqrt(1.0*n);   //忘记乘 1.0 CE了半天。。。
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            b[i] = a[i];
        }
        sort(b+1,b+n+1);
        int k = unique(b+1,b+n+1)-b-1;   //离散化
        for(int i=1;i<=n;i++)
            a[i] = lower_bound(b+1,b+k+1,a[i])-b;

        scanf("%d",&m);
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d",&t[i].l,&t[i].r);
            t[i].id = i;
            num[i] = (i-1)/block+1;
        }
        sort(t+1,t+m+1,cmp);
        l = t[1].l , r = t[1].l-1;
        ans = 0;
        for(int i=1;i<=m;i++)
        {
            while(l>t[i].l)  add(a[--l]);  //注意顺序
            while(r<t[i].r)  add(a[++r]);
            while(l<t[i].l)  del(a[l++]);
            while(r>t[i].r)  del(a[r--]);
            t[i].sum = ans;
        }
        sort(t+1,t+m+1,cmp2);
        for(int i=1;i<=m;i++)
            printf("%lld\n",t[i].sum);
    }
    return 0;
}