Codeforces 86D

题目链接：http://codeforces.com/problemset/problem/86/D

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

解题思路：莫队模板题，改一下add,remove就好了。

#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int N=2e5+5;
const int MAX=1e6+5;
int  cnt[MAX],arr[N],unit;
ll power,res[N];

struct node{
    int l,r;
    int id;
}q[N];

void add(int pos){
    power=power+(ll)arr[pos]*(cnt[arr[pos]]<<1|1);//(a+1)*(a+1)=a*a+2*a+1;
    cnt[arr[pos]]++;
}

void remove(int pos){    
    power=power-(ll)arr[pos]*(cnt[arr[pos]]<<1-1);
    cnt[arr[pos]]--;
}

bool cmp(node a,node b){
    return a.l/unit!=b.l/unit?a.l/unit<b.l/unit:a.r<b.r;
}

int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    unit=sqrt(n);
    for(int i=1;i<=n;i++){
        scanf("%d",&arr[i]);
    }
    for(int i=1;i<=m;i++){
        scanf("%d%d",&q[i].l,&q[i].r);
        q[i].id=i;
    }
    sort(q+1,q+1+m,cmp);
    
    int L=q[1].l,R=L-1;
    for(int 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--);
        res[q[i].id]=power;
    }
    for(int i=1;i<=m;i++){
        printf("%I64d
",res[i]);
    } 
}