U149858

題意

給定兩個長度爲 $n$ 的序列 $a,b$。 

對於一個 $1$ 到 $n$ 的排列 $p$,記 $c_i=\gcd(a_i,b_{p_i})$,$\sigma(c)$ 表示序列 $c$ 中所有元素的方差。 

求 $$\sum\limits_{p}\sigma(c)$$ 對 $10^9+7$ 取模。

$2\leq n,a_i,b_i\leq 10^6$ 。

題解 

方差可以寫成以下形式 

$$\begin{aligned}\sigma(c)&=\dfrac{1}{n}\sum_{i=1}^n (c_i-\bar{c})^2\\&=\dfrac{1}{n}\sum_{i=1}^n c_i^2-\dfrac{1}{n^2}(\sum_{i=1}^n c_i)^2\end{aligned}$$

那麼當前問題是求出 $\sum_p \sum_{i=1}^n c_i^2$ 與 $\sum_p (\sum_{i=1}^n c_i)^2$ ,先考慮左式。 

顯然可以考慮 $(a_i,b_j)$ 點對貢獻最後相加,那麼答案可以寫成

$$(n-1)!\times \sum_{i=1}^n\sum_{j=1}^n\gcd(A_i,B_j)\gcd(A_i,B_j)$$

對於右面的可以從大到小容斥得出有多少個點對的 $\gcd=k$ 。

故處理當前部分的時間複雜度是調和級數 $\mathcal O(m\log m)$ ,其中 $m$ 爲數的值域。

同理考慮右式子,由於平方故要考慮兩個點對的影響

$$(n-1)!\times \sum_{i=1}^n\sum_{j=1}^n\gcd(A_i,B_j)\gcd(A_i,B_j)\\(n-2)!\sum_{i=1}^n\sum_{i'=1}^n\sum_{j=1}^n\sum_{j'=1}^n [i\neq i'][j\neq j']\gcd(A_i,B_j)\gcd(A_{i'},B_{j'})$$

爲上面兩式之和,第一行在上面已經算出故我們僅用考慮第二行。

由於 $\neq $ 關係非常難受,對 $\neq $ 進行容斥

$$\sum_{i=1}^n\sum_{i'=1}^n\sum_{j=1}^n\sum_{j'=1}^n\gcd(A_i,B_j)\gcd(A_{i'},B_{j'})\\\sum_{i=1}^n\sum_{i'=1}^n\sum_{j=1}^n\sum_{j'=1}^n [i= i']\gcd(A_i,B_j)\gcd(A_{i'},B_{j'})\\\sum_{i=1}^n\sum_{i'=1}^n\sum_{j=1}^n\sum_{j'=1}^n [j= j']\gcd(A_i,B_j)\gcd(A_{i'},B_{j'})\\\sum_{i=1}^n\sum_{i'=1}^n\sum_{j=1}^n\sum_{j'=1}^n [i=i'][j= j']\gcd(A_i,B_j)\gcd(A_{i'},B_{j'})$$

其中第一個和第四個可以直接用算左式的結果,二三進行歐拉反演也可以做到 $\mathcal O(m\log m)$ .

總時間複雜度 $\mathcal O(m\log m)$ 。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
#include<climits>
#define pii pair<int,int>
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define int long long
#define mod 1000000007
using namespace std;
inline int read(){
    int f=1,ans=0;char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){ans=ans*10+c-'0';c=getchar();}
    return f*ans;
}
const int MAXN=1e6+11;
int v[MAXN],pri[MAXN],phi[MAXN],N,A[MAXN],B[MAXN],bucA[MAXN],bucB[MAXN],Lim=1000000,Ans1,Ans2;
int BucA[MAXN],BucB[MAXN],f[MAXN],fac[MAXN];
void init(){
    phi[1]=1; for(int i=2;i<=Lim;i++){
        if(!v[i]){pri[++pri[0]]=i,v[i]=i,phi[i]=i-1;}
        for(int j=1;j<=pri[0]&&pri[j]*i<=Lim;j++){
            v[i*pri[j]]=pri[j]; if(!(i%pri[j])){phi[i*pri[j]]=phi[i]*pri[j];break;}
            phi[i*pri[j]]=phi[i]*(pri[j]-1);
        }
    }return;
}
int gcd(int a,int b){return !b?a:gcd(b,a%b);}
int ksm(int a,int b){int ans=1;while(b){if(b&1) ans*=a,ans%=mod;a*=a,a%=mod;b>>=1;}return ans;}
int res1,res2,res3,res4,S1[MAXN],S2[MAXN];
signed main(){
    //freopen("B.in","r",stdin);
    N=read(); for(int i=1;i<=N;i++) A[i]=read(),bucA[A[i]]++; for(int i=1;i<=N;i++) B[i]=read(),bucB[B[i]]++; init();
    fac[0]=1; for(int i=1;i<=N;i++) fac[i]=fac[i-1]*i%mod;
    for(int d=1;d<=Lim;d++) for(int i=d;i<=Lim;i+=d) BucA[d]+=bucA[i],BucB[d]+=bucB[i];
    for(int i=1;i<=Lim;i++) f[i]=BucA[i]*BucB[i]%mod;
    for(int i=Lim;i>=1;i--) for(int j=2*i;j<=Lim;j+=i) f[i]=(f[i]-f[j]+mod)%mod;
    for(int i=1;i<=Lim;i++) Ans1+=f[i]*i%mod*i%mod,Ans1%=mod; res4=Ans1; Ans1*=fac[N-1],Ans1%=mod;
    Ans2=Ans1;
    for(int i=1;i<=Lim;i++) res1+=f[i]*i%mod,res1%=mod; res1=res1*res1%mod;
    for(int i=1;i<=Lim;i++){
        int w1=phi[i]*BucB[i]%mod,w2=phi[i]*BucA[i]%mod;
        for(int j=i;j<=Lim;j+=i) S1[j]+=w1,S1[j]%=mod,S2[j]+=w2,S2[j]%=mod;
    }
    for(int i=1;i<=N;i++) res2+=S1[A[i]]*S1[A[i]]%mod,res2%=mod;
    for(int i=1;i<=N;i++) res3+=S2[B[i]]*S2[B[i]]%mod,res3%=mod;
    Ans2+=(res1+res4-res2-res3)*fac[N-2]%mod,Ans2=(Ans2%mod+mod)%mod;
    //cerr<<Ans1<<" "<<Ans2<<endl;
    printf("%lld\n",(Ans1*ksm(N,mod-2)%mod-Ans2*ksm(N*N%mod,mod-2)%mod+mod)%mod);
}/*
  3
  1 2 3
  1 2 3
  */
View Code

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章