题面:https://www.luogu.org/problemnew/show/P2709
本题的关键在于sum(n)²-sum(n-1)²=sum(n)²-sum(n)²+2*sum(n)-1=2*sum(n)-1
Code:
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<ctime>
using namespace std;
const int N=50005;
int c[N],sum[N],anss[N],block,ans;
struct Node{
int l,r,num;
}q[N];
bool cmp(Node a,Node b){
return (a.l/block)^(b.l/block)?a.l<b.l:(((a.l/block)&1)?a.r<b.r:a.r>b.r);
}
void del(int x){
sum[c[x]]--;
ans-=2*sum[c[x]]+1;
}
void add(int x){
sum[c[x]]++;
ans+=2*sum[c[x]]-1;
}
int main(){
int n,m,k;
scanf("%d%d%d",&n,&m,&k);
block=sqrt(n);
for(int i=1;i<=n;i++){
scanf("%d",&c[i]);
}
for(int i=1;i<=m;i++){
scanf("%d%d",&q[i].l,&q[i].r);
q[i].num=i;
}
sort(q+1,q+1+m,cmp);
int l=1,r=0;
for(int i=1;i<=m;i++){
int ql=q[i].l,qr=q[i].r;
while(l>ql){
l--;
add(l);
}
while(r<qr){
r++;
add(r);
}
while(l<ql){
del(l);
l++;
}
while(r>qr){
del(r);
r--;
}
anss[q[i].num]=ans;
}
for(int i=1;i<=m;i++){
printf("%d\n",anss[i]);
}
return 0;
}