Sona NBUT - 1457 （莫队+hash）

题目：点击此处

题意：求一个区间内出现的数的次数的立方的和

思路：主要不同就是add和remove里的不同

add：当前某个数的出现次数是cnt，如果遇见了这个数，那么先让ans-=cnt*cnt*cnt，然后ans+=(cnt+1)*(cnt+1)*(cnt+1);

remove:当前某个数的出现次数是cnt，如果遇见了这个数，那么先让ans-=cnt*cnt*cnt，然后ans+=(cnt-1)*(cnt-1)*(cnt-1);

由于给的n个数字比较大，n比较小，所以必须离散化

注意不能用map直接操作，这样会超时

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <queue>
#include <vector>
#include <list>
#include <map>
#include <set>
#define ll long long
using namespace std;
const int maxn=1e5+5;
int n,q,a[maxn],block,l,r,b[maxn];
ll cnt[maxn],answer[maxn],ans;
map<int,int> mp;
struct node
{
    int l,r,i;
}pos[maxn];
bool cmp(node a,node b)
{
    if(a.l/block!=b.l/block)
        return a.l/block<b.l/block;
    return a.r<b.r;
}
ll cube(ll x)
{
    return x*x*x;
}
void add(int x)
{
    ans-=cube(cnt[a[x]]);
    cnt[a[x]]++;
    ans+=cube(cnt[a[x]]);
}
void remove1(int x)
{
    ans-=cube(cnt[a[x]]);
    cnt[a[x]]--;
    ans+=cube(cnt[a[x]]);
}
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]),b[i]=a[i];
        sort(b+1,b+n+1);
        memset(cnt,0,sizeof(cnt));
        int m=unique(b+1,b+1+n)-b-1;
        ans=0;
        for(int i=1;i<=n;i++)
        {
            a[i]=lower_bound(b+1,b+1+n,a[i])-b-1;
        }
        block=sqrt(1.0*m);
        scanf("%d",&q);
        for(int i=1;i<=q;i++)
        {
            scanf("%d%d",&pos[i].l,&pos[i].r);
            pos[i].i=i;
        }
        sort(pos+1,pos+1+q,cmp);
        ll cl=pos[1].l,cr=cl-1;
        for(int i=1;i<=q;i++)
        {
            l=pos[i].l,r=pos[i].r;
            while(cl<l)
            {
                remove1(cl++);
            }
            while(cl>l)
            {
                add(--cl);
            }
            while(cr<r)
            {
                add(++cr);
            }
            while(cr>r)
            {
                remove1(cr--);
            }
            answer[pos[i].i]=ans;
        }
        for(int i=1;i<=q;i++)
            printf("%lld\n",answer[i]);
    }
    return 0;
}