bzoj 3781: 小B的询问

最新推荐文章于 2022-07-18 06:27:57 发布

lqybzx

最新推荐文章于 2022-07-18 06:27:57 发布

阅读量815

点赞数

分类专栏：莫队算法文章标签：莫队算法

本文链接：https://blog.csdn.net/lqybzx/article/details/41779797

版权

莫队算法专栏收录该内容

9 篇文章 1 订阅

订阅专栏

Description

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

Input

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

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

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

Output

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

Sample Input

6 4 3
1 3 2 1 1 3
1 4
2 6
3 5
5 6

Sample Output

6
9
5
2

HINT

对于全部的数据，1<=N、M、K<=50000

同小Z的袜子。。莫队乱搞

#include<cmath>
#include<cstdio>
#include<algorithm>
using namespace std;
struct question
{
     int l,r;
     long long a,b;
     int x;
}ask[50001];
int a[50001],s[50001];
int pos[50001];
long long ans;
inline bool cmp1(question x,question y)
{
     if(pos[x.l]==pos[y.l])
     {
          if(x.r<y.r)
               return true;
          return false;
     }
     if(x.l<y.l)
          return true;
     return false;
}
inline bool cmp2(question x,question y)
{
     if(x.x<y.x)
          return true;
     return false;
}
inline void update(int x,int xx)
{
     ans-=s[a[x]]*s[a[x]];
     s[a[x]]+=xx;
     ans+=s[a[x]]*s[a[x]];
}
int main()
{
     int n,m,k;
     scanf("%d%d%d",&n,&m,&k);
     int i;
     int space=sqrt(n);
     for(i=1;i<=n;i++)
          scanf("%d",&a[i]);
     for(i=1;i<=m;i++)
     {
          scanf("%d%d",&ask[i].l,&ask[i].r);
          ask[i].x=i;
     }
     for(i=1;i<=n;i++)
          pos[i]=(i-1)/space+1;
     sort(ask+1,ask+1+m,cmp1);
     int l=1,r=0;
     for(i=1;i<=m;i++)
     {
          while(r<ask[i].r)
          {
               r++;
               update(r,1);
          }
          while(r>ask[i].r)
          {
               update(r,-1);
               r--;
          }
          while(l<ask[i].l)
          {
               update(l,-1);
               l++;
          }
          while(l>ask[i].l)
          {
               l--;
               update(l,1);
          }
          ask[i].a=ans;
     }
     sort(ask+1,ask+1+m,cmp2);
     for(i=1;i<=m;i++)
          printf("%lld\n",ask[i].a);
     return 0;
}