題目描述

小B 有一個長為nn的整數序列aa，值域為[1,k][1,k]。
他一共有mm個詢問，每個詢問給定一個區間[l,r][l,r]，求：

\sum\limits_{i=1}^k c_i^2i=1∑k​ci2​

其中c_ici​表示數字ii在[l,r][l,r]中的出現次數。
小B請你幫助他回答詢問。

輸入格式

第一行三個整數n,m,kn,m,k。

第二行nn個整數，表示 小B 的序列。

接下來的mm行，每行兩個整數l,rl,r。

輸出格式

輸出mm行，每行一個整數，對應一個詢問的答案。

題解：

#include<bits/stdc++.h>
using namespace
 
 std;
typedef long long ll;
const int maxn=2e5+100;
ll a[maxn];
ll cnt[maxn];
ll belong[maxn];
ll n,m,k,size,bnum,now,ans[maxn];



struct query {
    ll l,r,id;
}q[maxn];
int cmp (query a,query b) {
    return (belong[a.l]^belong[b.l])?belong[a.l]<belong[b.l]:((belong[a.l]&1)?a.r<b.r:a.r>b.r);
}
 
void add (int p) {
    //if (!cnt[a[p]])
    //    ++now;
    ++cnt[a[p]];
    now+=2*cnt[a[p]]-1;
}
void del (int p) {
    --cnt[a[p]];
    now-=2*cnt[a[p]]+1;
    //if (!cnt[a[p]])
    //    --now;
}
int main () {
    scanf("%lld%lld%lld",&n,&m,&k);
    size=sqrt(n);
    bnum=n/size+(n%size>0
 
);
    for (int i=1;i<=bnum;i++)
        for (int j=(i-1)*size+1;j<=i*size;j++)
            belong[j]=i;
    for (int i=1;i<=n;i++) scanf("%lld",a+i);
    for (int i=1;i<=m;i++) {
        scanf("%lld%lld",&q[i].l,&q[i].r);
        q[i].id=i;
    }
    sort(q+1,q+m+1,cmp);
    ll l=1,r=0;
    for (int i=1;i<=m;i++) {
        int ql=q[i].l;
        int qr=q[i].r;
        while (l<ql) del(l++);
        while (l>ql) add(--l);
        while (r<qr) add(++r);
        while (r>qr) del(r--); 
        ans[q[i].id]=now; 
    }
    for (int i=1;i<=m;i++) printf("%lld\n",ans[i]);
}