Evanyou Blog 彩带

P2709 小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个询问的答案。

输入输出样例

输入样例#1: 
  1. 6 4 3
  2. 1 3 2 1 1 3
  3. 1 4
  4. 2 6
  5. 3 5
  6. 5 6
输出样例#1: 
  1. 6
  2. 9
  3. 5
  4. 2

说明

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

 

Solution:

  本题是一道莫队模板题。

  直接由题意,维护每种颜色数的平方,然后就是莫队的分块->块内排序->指针移动->记录答案。

  注意一下排序时先判断左端所在块是否相同,相同就按右段升序排列,否则就按左端升序排列。

  本题就这么简单的$AC$了

代码:

 

复制代码
  1. #include<bits/stdc++.h>
  2. #define il inline
  3. #define ll long long
  4. using namespace std;
  5. const int N=100005;
  6. int n,m,k,pos[N],a[N],s[N],ans[N],tot;
  7. struct data{
  8. int l,r,id;
  9. }t[N];
  10. il int gi()
  11. {
  12. int a=0;char x=getchar();bool f=0;
  13. while((x<'0'||x>'9')&&x!='-')x=getchar();
  14. if(x=='-')x=getchar(),f=1;
  15. while(x>='0'&&x<='9')a=a*10+x-48,x=getchar();
  16. return f?-a:a;
  17. }
  18. il bool cmp(data a,data b){return pos[a.l]==pos[b.l]?a.r<b.r:a.l<b.l;}
  19. il void change(int k,int u)
  20. {
  21. tot-=s[a[k]]*s[a[k]];
  22. s[a[k]]+=u;
  23. tot+=s[a[k]]*s[a[k]];
  24. }
  25. int main()
  26. {
  27. n=gi(),m=gi(),k=gi();
  28. int ss=int(sqrt(n));
  29. for(int i=1;i<=n;i++)a[i]=gi(),pos[i]=(i-1)/ss+1;
  30. for(int i=1;i<=m;i++)t[i].l=gi(),t[i].r=gi(),t[i].id=i;
  31. sort(t+1,t+m+1,cmp);
  32. for(int i=1,l=1,r=0;i<=m;i++){
  33. while(r<t[i].r)change(r+1,1),r++;
  34. while(r>t[i].r)change(r,-1),r--;
  35. while(l>t[i].l)change(l-1,1),l--;
  36. while(l<t[i].l)change(l,-1),l++;
  37. ans[t[i].id]=tot;
  38. }
  39. for(int i=1;i<=m;i++)printf("%d\n",ans[i]);
  40. return 0;
  41. }
复制代码

 

 

 

 

 

posted @   five20  阅读(122)  评论(0编辑  收藏  举报
Live2D
欢迎阅读『P2709 小B的询问』
点击右上角即可分享
微信分享提示