3781: 小B的询问

题目链接

题目大意：有n个1~k之间的整数，m个询问，每个询问给定一个区间[L..R]，求Sigma(c(i)^2)的值，i的值从1到K，其中c(i)表示数字i在[L..R]中的重复次数。

题解：莫队搞搞

我的收获：抄抄板子

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define M 50500

int n,m,blo,now,k;
int a[M],pos[M],cnt[M],ans[M];

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

bool cmp(node x,node y){return pos[x.l]<pos[y.l]||(pos[x.l]==pos[y.l]&&x.r<y.r);}

void update(int x,int v)
{
    now-=cnt[x]*cnt[x];
    cnt[x]+=v;
    now+=cnt[x]*cnt[x];
}

void work()
{
    int l=1,r=0;
    for(int i=1;i<=m;i++)
    {
        while(r<q[i].r) update(a[++r],1);
        while(l>q[i].l) update(a[--l],1);
        while(r>q[i].r) update(a[r--],-1);
        while(l<q[i].l) update(a[l++],-1);
        ans[q[i].id]=now;
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
}

void init()
{
    cin>>n>>m>>k;blo=sqrt(n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]),pos[i]=(i-1)/blo+1;
    for(int i=1;i<=m;i++)
        scanf("%d%d",&q[i].l,&q[i].r),q[i].id=i;
    sort(q+1,q+m+1,cmp);
}

int main()
{
    init();
    work();
    return 0;
}