题目链接:https://www.luogu.com.cn/problem/P2709
这道题是模板莫队,然后$i$在$[l,r]$区间内的个数就是$vis[ ]$数组
$add()$和$del()$的话就是先减去原来位置数的个数的平方,然后再加上现在位置数的个数的平方。
AC代码:
#include<cstdio>
#include<iostream>
#include<cmath>
#include<algorithm> using namespace std;
const int N=; int a[N],vis[N],ans,tot[N];
int block; struct node{
int l,r;
int id;
}q[N]; bool cmp(node aa,node bb){
if(aa.l/block==bb.l/block) return aa.r<bb.r;
return aa.l/block<bb.l/block;
} void add(int pos){
ans-=vis[a[pos]]*vis[a[pos]];
vis[a[pos]]++;
ans+=vis[a[pos]]*vis[a[pos]];
} void del(int pos){
ans-=vis[a[pos]]*vis[a[pos]];
vis[a[pos]]--;
ans+=vis[a[pos]]*vis[a[pos]];
} int main(){
int n,m,k;
scanf("%d%d%d",&n,&m,&k);
block=sqrt(n);
for(int i=;i<=n;i++) scanf("%d",&a[i]);
for(int i=;i<=m;i++){
scanf("%d%d",&q[i].l,&q[i].r);
q[i].id=i;
}
sort(q+,q+m+,cmp);
int L=,R=;
for(int i=;i<=m;i++){
while(L>q[i].l){
L--;
add(L);
}
while(L<q[i].l){
del(L);
L++;
}
while(R<q[i].r){
R++;
add(R);
}
while(R>q[i].r){
del(R);
R--;
}
tot[q[i].id]=ans;
}
for(int i=;i<=m;i++) printf("%d\n",tot[i]);
return ;
}
AC代码