codeforces 86D. Powerful array 莫队算法 模板题

题目大意：给定一个数组，每次询问一个区间[l,r]，设cnt[i]为数字i在该区间内的出现次数，求该区间内所有的cnt[i]^2 *i。

 

打多校碰到的莫队算法 学习了一下 第一个模板题~

#include <cstdio>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <vector>
#include <iostream>
#define ll long long
#define INF 0x3f3f3f
using namespace std;
const int N=400000+100;
int a[N],unit,cnt[N*3];
ll ans[N];
ll power;
struct node
{
    int l,r,id;
}q[N];
bool cmp(node a,node b)
{
    if(a.l/unit!=b.l/unit) return a.l<b.l;
    return a.r<b.r;
}
void add(int pos)
{
    power=power+(ll)a[pos]*((ll)2*cnt[a[pos]]+(ll)1);
    cnt[a[pos]]++;
}
void remove(int pos)
{
    power=power+(ll)a[pos]*((ll)1-(ll)2*cnt[a[pos]]);
    cnt[a[pos]]--;
}
int main()
{
    int n,m,i,L,R;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++) scanf("%d",&a[i]);
    for(i=1;i<=m;i++)
    {
        scanf("%d%d",&q[i].l,&q[i].r);
        q[i].id=i;
    }
    unit=sqrt(n);
    sort(q+1,q+1+m,cmp);
    L=q[1].l;R=L-1;
    for(i=1;i<=m;i++)
    {
        while(L>q[i].l) add(--L);
        while(L<q[i].l) remove(L++);
        while(R<q[i].r) add(++R);
        while(R>q[i].r) remove(R--);
        ans[q[i].id]=power;
    }
    for(i=1;i<=m;i++) printf("%I64d\n",ans[i]);
    return 0;
}