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:
- 6 4 3
- 1 3 2 1 1 3
- 1 4
- 2 6
- 3 5
- 5 6
输出样例#1:
- 6
- 9
- 5
- 2
说明
对于全部的数据,1<=N、M、K<=50000
Solution:
本题是一道莫队模板题。
直接由题意,维护每种颜色数的平方,然后就是莫队的分块->块内排序->指针移动->记录答案。
注意一下排序时先判断左端所在块是否相同,相同就按右段升序排列,否则就按左端升序排列。
本题就这么简单的$AC$了
代码:
- #include<bits/stdc++.h>
- #define il inline
- #define ll long long
- using namespace std;
- const int N=100005;
- int n,m,k,pos[N],a[N],s[N],ans[N],tot;
- struct data{
- int l,r,id;
- }t[N];
- il int gi()
- {
- int a=0;char x=getchar();bool f=0;
- while((x<'0'||x>'9')&&x!='-')x=getchar();
- if(x=='-')x=getchar(),f=1;
- while(x>='0'&&x<='9')a=a*10+x-48,x=getchar();
- return f?-a:a;
- }
- il bool cmp(data a,data b){return pos[a.l]==pos[b.l]?a.r<b.r:a.l<b.l;}
- il void change(int k,int u)
- {
- tot-=s[a[k]]*s[a[k]];
- s[a[k]]+=u;
- tot+=s[a[k]]*s[a[k]];
- }
- int main()
- {
- n=gi(),m=gi(),k=gi();
- int ss=int(sqrt(n));
- for(int i=1;i<=n;i++)a[i]=gi(),pos[i]=(i-1)/ss+1;
- for(int i=1;i<=m;i++)t[i].l=gi(),t[i].r=gi(),t[i].id=i;
- sort(t+1,t+m+1,cmp);
- for(int i=1,l=1,r=0;i<=m;i++){
- while(r<t[i].r)change(r+1,1),r++;
- while(r>t[i].r)change(r,-1),r--;
- while(l>t[i].l)change(l-1,1),l--;
- while(l<t[i].l)change(l,-1),l++;
- ans[t[i].id]=tot;
- }
- for(int i=1;i<=m;i++)printf("%d\n",ans[i]);
- return 0;
- }
PS:~蒟蒻写博客不易,转载请注明出处,万分感谢!~
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步