NBUT - 1457(Sona)（莫队+离散化）

题意：求查询区间的各数字个数的立方和。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <math.h>
using namespace std;
typedef long long ll;
//思路(离散化+莫队)：由于数组开不了10^9的空间,先对数据进行离散化,在对询问用莫队离线处理;
//为了不改变原数字序列数字之间的相对位置,用另一个数组copy原数组进行离散化;
const int N=100050;
int a[N],c[N];
ll cnt[N],res[N];
ll ans;
int unit;
struct node
{
    int l,r,id;
}query[N];

bool cmp(node a, node b)
{
    if(a.l/unit==b.l/unit)
    {
        return a.r<b.r;
    }
    else
    {
        return a.l<b.l;
    }
}

inline int read()//此方法只能读入整数;
{
    int sgn = 1; int cnt = 0;   //sgn表示正负号 cnt表示读取的数字
    char ch = getchar();
    while (ch < '0' || ch > '9') {      //读取非数字的字符，保留负号，忽略其余无关符号
        if(ch == '-')
            sgn = -sgn;
        ch = getchar();
    }
    while ('0' <= ch && ch <= '9') {
        cnt = cnt*10 + (ch-'0');
        ch = getchar();
    }
    return  sgn*cnt;
}

void add(int pos)
{
    ans-=cnt[a[pos]]*cnt[a[pos]]*cnt[a[pos]];
    cnt[a[pos]]++;
    ans+=cnt[a[pos]]*cnt[a[pos]]*cnt[a[pos]];
}

void remove(int pos)
{
    ans-=cnt[a[pos]]*cnt[a[pos]]*cnt[a[pos]];
    cnt[a[pos]]--;
    ans+=cnt[a[pos]]*cnt[a[pos]]*cnt[a[pos]];
}

int main()
{
    int i,j;
    int n,m;
    while(scanf("%d",&n)!=EOF)
    {
        unit=sqrt(n);
        for(i=1;i<=n;i++)
        {
            a[i]=read();
            c[i]=a[i];
        }
        sort(c+1,c+1+n);
        int size=unique(c+1,c+1+n)-(c+1);
        for(i=1;i<=n;i++)
        {
            a[i]=lower_bound(c+1,c+1+size,a[i])-c;
        }
        memset(cnt,0,sizeof(cnt));
        m=read();
        for(i=1;i<=m;i++)
        {
            query[i].l=read();
            query[i].r=read();
            query[i].id=i;
        }
        sort(query+1,query+1+m,cmp);
        ans=0;
        int L=query[1].l;
        int R=L-1;
        for(i=1;i<=m;i++)
        {
            while(L>query[i].l)
            {
                add(--L);
            }
            while(L<query[i].l)
            {
                remove(L++);
            }
            while(R>query[i].r)
            {
                remove(R--);
            }
            while(R<query[i].r)
            {
                add(++R);
            }
            res[query[i].id]=ans;
        }
        for(i=1;i<=m;i++)
        {
            printf("%I64d\n",res[i]);
        }
    }
    return 0;
}