數據範圍很容易讓人想到莫隊算法
但是對於每次詢問有\(l_1,r_1,l_2,r_2\)四個參數
很不方便維護
所以可以將詢問差分

\(get(l,r,x)=get(1,r,x)-get(1,l-1,x)\)
\(get(l_1,r_1,x)*get(l_2,r_2,x)\)
\(=get(1,r_1,x)*get(1,r_2,x)\)
\(-get(1,l_1-1,x)*get(1,r_2,x)\)
\(-get(1,r_1,x)*get(1,l_2-1,x)\)
\(+get(1,l_1-1,x)*get(1,l_2-1,x)\)

於是原問題轉化為四個只有兩個參數的問題且可以使用莫隊求解的問題

#include<bits/stdc++.h>

using namespace std;

#define gc c=getchar()
#define r(x) read(x)
#define ll long long

template<typename T>
inline void read(T&x){
    x=0;T k=1;char gc;
    while(!isdigit(c)){if(c==‘-‘)k=-1;gc;}
    while(isdigit(c)){x=x*10+c-‘0‘;gc;}x*=k;
}

const int N=50005;

int be[N],a[N];

struct Query{
    int l,r,type,id;
    bool operator < (Query x) const{
        return be[l]==be[x.l]?r<x.r:l<x.l;
    }
};

ll Ans[N];
vector<Query> Q;

int cntl[N],cntr[N];

int main(){
    int n,m;r(n);
    int block_size=sqrt(n);
    for(int i=1;i<=n;++i){
        r(a[i]);
        be[i]=(i-1)/block_size+1;
    }
    r(m);
    for(int i=1,l1,r1,l2,r2;i<=m;++i){
        r(l1),r(r1),r(l2),r(r2);
        Q.push_back(Query{r1,r2,1,i});
        if(l1>1)Q.push_back(Query{l1-1,r2,-1,i});
        if(l2>1)Q.push_back(Query{r1,l2-1,-1,i});
        if(l1>1&&l2>1)Q.push_back(Query{l1-1,l2-1,1,i});
    }
    sort(Q.begin(),Q.end());
    ll ans=0;
    int l=0,r=0;
    for(int i=0;i<Q.size();++i){
        Query &x=Q[i];
        
        while(l<x.l)cntl[a[++l]]++,ans+=cntr[a[l]];
        while(l>x.l)cntl[a[l]]--,ans-=cntr[a[l--]];
        while(r<x.r)cntr[a[++r]]++,ans+=cntl[a[r]];
        while(r>x.r)cntr[a[r]]--,ans-=cntl[a[r--]];
        
        Ans[x.id]+=x.type*ans;
    }
    for(int i=1;i<=m;++i)printf("%lld\n",Ans[i]);
}
 

[Snoi2017]一個簡單的詢問