51nod1584加权约数和

题目大意：

求:

\[\sum_{i-1}^n\sum_{j=1}^nmax(i,j)\sigma(i*j)
\]

题解

对于这个\(\max\)，套路的把它转化成：

\[2*\sum_{i=1}^n\sum_{j=1}^ii*\sigma(i*j)-\sum_{i=1}^n i*\sigma(i*i)
\]

对于前面的部分，我们可以：

\[\sum_{i=1}^{n}\sum_{j=1}^ii\sum_{a|i}\sum_{b|j}a*\frac{j}{b}[(a,b)==1]
\]

\[\sum_{i=1}^{n}i\sum_{j=i}^n\sum_{a|i}\sum_{b|j}a*\frac{j}{b}\sum_{d|(i,j)}\mu(d)
\]

\[\sum_{d=1}^n\mu(d)\sum_{i=1}^{\frac{n}{d}}i*d\sum_{a|i}a*d\sum_{j=1}^{i}\sum_{b|j}\frac{j}{b}
\]

\[\sum_{d=1}^n\mu(d)d^2\sum_{i=1}^{\frac{n}{d}}i\sum_{a|i}a\sum_{j=1}^i\sum_{b|j}\frac{j}{b}
\]

\[\sum_{d=1}^n\mu(d)d^2\sum_{i=1}^{\frac{n}{d}}g_i
\]

\[\sum_{D=1}^n\sum_{d|D}\mu(d)d^2G_{D/d}
\]

\[g_n=n*\sigma_n*\sum_{i=1}^n \sigma_i
\]

这个\(g\)数组就可以线性预处理了。

后面的部分可以线性筛，姿势++。

代码

#include<bits/stdc++.h>

#define N 1000009

using namespace std;

typedef long long ll;

const int maxn=1000000;

const int mod=1000000007;

bool vis[N];

int prime[N];

ll mu[N],md[N],mdp[N],ans[N],g[N],sum[N],f[N];

ll sig[N],sig2[N];

inline ll rd(){

	ll x=0;char c=getchar();bool f=0;

	while(!isdigit(c)){if(c=='-')f=1;c=getchar();}

	while(isdigit(c)){x=(x<<1)+(x<<3)+(c^48);c=getchar();}

	return f?-x:x;

}

inline void MOD(ll &x){x=x>=mod?x-mod:x;}

inline void prework(int n){

	sig[1]=mu[1]=sig2[1]=md[1]=mdp[1]=1;

	for(int i=2;i<=n;++i){

		//cout<<i<<" "<<md[i]<<" "<<mdp[i]<<endl;

		if(!vis[i]){

			prime[++prime[0]]=i;

			md[i]=mdp[i]=i;

			mu[i]=mod-1;

			sig[i]=i+1;

			sig2[i]=(1ll*i*i%mod+i+1)%mod;

		}

		for(int j=1;j<=prime[0]&&(i*prime[j])<=n;++j){

			vis[i*prime[j]]=1;

			if(i%prime[j]==0){

				mu[i*prime[j]]=0;

				md[i*prime[j]]=prime[j];

				mdp[i*prime[j]]=mdp[i]*prime[j];

				sig[i*prime[j]]=(sig[i]+1ll*prime[j]*mdp[i]%mod*sig[i/mdp[i]]%mod)%mod;

                sig2[i*prime[j]]=sig2[i]+(1ll*mdp[i]*mdp[i]%mod*md[i]%mod+

                1ll*mdp[i]*mdp[i]%mod*md[i]%mod*md[i]%mod)*sig2[i/mdp[i]]%mod;

                sig2[i*prime[j]]%=mod;

				break;

			}

			mu[i*prime[j]]=mod-mu[i];

            sig[i*prime[j]]=sig[i]*sig[prime[j]]%mod;

            md[i*prime[j]]=mdp[i*prime[j]]=prime[j];

            sig2[i*prime[j]]=sig2[i]*sig2[prime[j]]%mod;

		}

	}

	for(int i=1;i<=n;++i)MOD(sum[i]=sum[i-1]+sig[i]);

	for(int i=1;i<=n;++i){

        g[i]=1ll*sig[i]*i%mod*sum[i]%mod;

        MOD(sig2[i]=sig2[i-1]+sig2[i]*i%mod);

        for(int j=i;j<=n;j+=i)MOD(f[j]+=g[i]*mu[j/i]%mod*(j/i)%mod*(j/i)%mod);

        MOD(f[i]+=f[i-1]);

		ans[i]=(f[i]*2-sig2[i]+mod)%mod;

	}

}

int main(){

	prework(maxn);

	int T=rd(),ct=0;

	while(T--){

      int x=rd();ct++;

      printf("Case #%d: %lld\n",ct,ans[x]);

	}

	return 0;

}

巴特西

51nod1584加权约数和

题目大意：

题解

代码

最新文章

热门文章