小B的询问

这题跟项链那道差不多，也是莫队。只是计数方式不一样
QAQ
这个是让统计区间里出现的数字出现次数平方的总和。

#include <cstdio>
#include <iostream>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
struct s{
    int l;
    int r;
    int id;
    int k;
    int ss;
}ask[1000001];
int a[1000001];
int num[1000001];
int ans[1000001];
int sum,l,r;
int comp(const s&a,const s&b)
{
    if(a.k!=b.k)
     return a.k<b.k;
    return a.r<b.r;
}
void add(int x)
{
    sum+=2*num[x]+1;//（x+1）的平方=x的平方+2*x+1。（很简单的数学23333）
    num[x]++;
} 
void del(int x)
{
    sum+=1;
    sum-=2*num[x];//（x-1）的平方=x的平方-2*x+1。（很简单的数学23333）
    num[x]--;
}
int main()
{
    int n,m,k;

    scanf("%d%d%d",&n,&m,&k);

    int n1=sqrt(double(n));
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    for(int i=1;i<=m;i++)
     {
         scanf("%d%d",&ask[i].l,&ask[i].r);
         ask[i].id=i;
         ask[i].k=ask[i].l/n1; 
     } 
    sort(ask+1,ask+m+1,comp);

    for(int i=1;i<=m;i++)
    {
        while(r<ask[i].r)
       {
          r++;
          add(a[r]);
       }
      while(r>ask[i].r)
      {
         del(a[r]);
         r--;
      }
      while(l<ask[i].l)
       {
           del(a[l]);
          l++;
       }
      while(l>ask[i].l)
      { 
         l--;
         add(a[l]);
      }
      ans[ask[i].id]=sum-1;
    }

     for(int i=1;i<=m;i++)
      printf("%d\n",ans[i]);

    return 0;
}