NBUT 1457 Sona (莫队算法)

题意：

给出一个数列，求区间内相同数字次数的三次方和

分析：

离散化数字之后直接莫队

#include<cstring>
#include<string>
#include<iostream>
#include<queue>
#include<cstdio>
#include<algorithm>
#include<map>
#include<cstdlib>
#include<cmath>
#include<vector>
//#pragma comment(linker, "/STACK:1024000000,1024000000");

using namespace std;

#define INF 0x3f3f3f3f
#define maxn 100005

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

vector<int>v;
long long ans[maxn];
int vis[maxn];
int a[maxn];
int block[maxn];
long long sum;

bool cmp(node A,node B)
{
    if(block[A.l]==block[B.l]) return A.r<B.r;
    return block[A.l]<block[B.l];
}

long long cal(long long x)
{
    return 3*x*x+3*x+1;
}

void update(int add,int note)
{
    if(add==-1) sum-=cal(vis[note]-1),vis[note]--;
    else sum+=cal(vis[note]),vis[note]++;
}

int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        memset(vis,0,sizeof vis);
        v.clear();
        int bk=sqrt(n*1.0);
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
            v.push_back(a[i]);
            block[i]=(i/bk)+1;
        }
        block[n+1]=block[n]+1;
        sort(v.begin(),v.end());
        v.erase(unique(v.begin(),v.end()),v.end());
        for(int i=1; i<=n; i++)
        {
            a[i]=lower_bound(v.begin(),v.end(),a[i])-v.begin();
            a[i]++;
        }
        int m;
        scanf("%d",&m);
        for(int i=0; i<m; i++)
        {
            scanf("%d%d",&q[i].l,&q[i].r);
            q[i].id=i;
        }
        sort(q,q+m,cmp);
        int l=0,r=1;
        sum=0;
        for(int i=0; i<m; i++)
        {
            for(; r<=q[i].r; r++) update(1,a[r]);
            for(; r-1>q[i].r; r--) update(-1,a[r-1]);
            for(; l+1<q[i].l; l++) update(-1,a[l+1]);
            for(; l>=q[i].l; l--) update(1,a[l]);
            ans[q[i].id]=sum;
        }
        for(int i=0; i<m; i++)
        {
            printf("%I64d\n",ans[i]);
        }
    }
    return 0;
}