小B的询问（莫队）

题目描述

小B有一个序列，包含N个1~K之间的整数。他一共有M个询问，每个询问给定一个区间[L..R]，求Sigma(c(i)^2)的值,其中i的值从1到K，其中c(i)表示数字i在[L..R]中的重复次数。小B请你帮助他回答询问。

输入输出格式

输入格式：

第一行，三个整数N、M、K。

第二行，N个整数，表示小B的序列。

接下来的M行，每行两个整数L、R。

输出格式：

M行，每行一个整数，其中第i行的整数表示第i个询问的答案。

思路：

显然，又是一道莫队裸题

颜色只有50000种

那么我们开一个50000的数组存下每种颜色的出现次数

跑莫队即可

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#define rii register int i
using namespace std;
int xl[50005],sy[50005],len,n,m,k,sum[500005],ans;
struct query{
    int l,r,bh,ans;
}x[50005];
bool cmp(query lk,query kl)
{
    if(sy[lk.l]==sy[kl.r])
    {
        return lk.r<kl.r;
    }
    return lk.l<kl.l;
}
void change(int wz,int val)
{
    ans-=sum[xl[wz]]*sum[xl[wz]];
    sum[xl[wz]]+=val;
    ans+=sum[xl[wz]]*sum[xl[wz]];
}
bool cmp1(query lk,query kl)
{
    return lk.bh<kl.bh;
}
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    len=sqrt(n);
    for(rii=1;i<=n;i++)
    {
        scanf("%d",&xl[i]);
        sy[i]=i/len+1;
    }
    for(rii=1;i<=m;i++)
    {
        scanf("%d%d",&x[i].l,&x[i].r);
        x[i].bh=i;
    }
    sort(x+1,x+m+1,cmp);
    int l=1,r=1;
    ans=1,sum[xl[1]]=1;
    for(rii=1;i<=m;i++)
    {
        while(l<x[i].l)
        {
            change(l,-1);
            l++;
        }
        while(l>x[i].l)
        {            

            change(l-1,1);
            l--;
        }
        while(r>x[i].r)
        {
            change(r,-1);
            r--;
        }
        while(r<x[i].r)
        {
            change(r+1,1);
            r++;
        }
        x[i].ans=ans;
    }            
    sort(x+1,x+m+1,cmp1);
    for(rii=1;i<=m;i++)
    {
        printf("%d\n",x[i].ans);
    }
}