小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
#include<iostream> #include<cstdio> #include<algorithm> #include<cmath> using namespace std; const int N=50010; int number[N]; int answer[N]; int pos[N]; int a[N]; int n,m,k,ans; struct node{ int l,r,num; }E[N]; inline int read() { int x=0;char c=getchar(); while(c<‘0‘||c>‘9‘)c=getchar(); while(c>=‘0‘&&c<=‘9‘)x=x*10+c-‘0‘,c=getchar(); return x; } bool cmp(node a,node b) { if(pos[a.l]==pos[b.l]) return a.r<b.r; else return pos[a.l]<pos[b.l]; } int add(int x) { ans-=number[x]*number[x]; number[x]++; ans+=number[x]*number[x]; } int dale(int x) { ans-=number[x]*number[x]; number[x]--; ans+=number[x]*number[x]; } int MD() { int ll=1,rr=0; for(int i=1;i<=m;i++) { for(;E[i].l<ll;ll--)add(a[ll-1]); for(;E[i].l>ll;ll++)dale(a[ll]); for(;E[i].r>rr;rr++)add(a[rr+1]); for(;E[i].r<rr;rr--)dale(a[rr]); answer[E[i].num]=ans; } } int main() { n=read(),m=read(),k=read(); for(int i=1;i<=n;i++) a[i]=read(); for(int i=1;i<=m;i++) {E[i].l=read();E[i].r=read();E[i].num=i;} int base=sqrt(n); for(int i=1;i<=n;i++) pos[i]=(i-1)/base+1; sort(E+1,E+m+1,cmp); MD(); for(int i=1;i<=m;i++) printf("%d\n",answer[i]); return 0; }