51nod1584 加权约数和【线性筛】

题目描述：

求$$\sum _{i=1}^N\sum _{j=1}^N\max (i,j)\cdot \sigma (i\cdot j)\mathrm{m}\mathrm{o}\mathrm{d}10^9+7$$
其中$\sigma (i\cdot j)$表示$i\cdot j$的约数之和，多组数据，$N\le 1000000,T\le 50000$

题目分析：

$$\begin{array}{rl}& \sum _{i=1}^N\sum _{j=1}^N\max (i,j)\cdot \sigma (i\cdot j)\\ =& 2\left(\sum _{i=1}^{N}i\sum _{j=1}^i\sigma (i\cdot j)\right)-\sum _{i=1}^{N}i\cdot \sigma (i^2)\end{array}$$
记上式大括号里的式子为$F(N)$，减号后面的式子为$G(N)$

$$\begin{array}{rl}F(N)& =\sum _{i=1}^{N}i\sum _{j=1}^i\sigma (i\cdot j)\\ & =\sum _{i=1}^{N}i\sum _{j=1}^i\sum _{x|i}\sum _{y|j}[(x,y)==1]x\ast j/y\\ & =\sum _{k=1}^N\mu (k)\sum _{i=1}^{\lfloor \frac{N}{k}\rfloor }ik\sum _{j=1}^i\sum _{x|i}\sum _{y|j}xk\ast jk/(yk)\\ & =\sum _{k=1}^N\mu (k)k^2\sum _{i=1}^{\lfloor \frac{N}{k}\rfloor }i\sum _{j=1}^i\sum _{x|i}x\sum _{j|y}\frac{j}{y}\\ & =\sum _{k=1}^N\mu (k)k^2\sum _{i=1}^{\lfloor \frac{N}{k}\rfloor }i\cdot \sigma (i)\sum _{j=1}^i\sigma (j)\end{array}$$
看起来可以上杜教筛了，然而多组数据会T，N<=1000000，直接$O(n\ln n)$预处理然后$O(1)$回答询问反而是更好的方式：
$$F(N)=\sum _{k=1}^N\sum _{i|k}\mu (\frac{k}{i})\cdot (\frac{k}{i}{)}^2\cdot i\cdot \sigma (i)\sum _{j=1}^i\sigma (j)$$
枚举$i$对$k$做贡献然后前缀和一下就可以了。
$G(N)$是类似的处理方式，将$\sigma (i^2)$展开为${\sum }_{x|i}{\sum }_{y|i}[(x,y)==1]x\ast i/y$，最终可以得到：
$$G(N)=\sum _{k=1}^N\sum _{i|k}\mu (\frac{k}{i})\cdot (\frac{k}{i}{)}^2\cdot i\cdot {\sigma }^2(i)$$

Code：

#include<bits/stdc++.h>
#define maxn 1000005
using namespace std;
const int mod = 1e9+7;
int T,n,p[maxn/8],mu[maxn],sig[maxn],mps[maxn],ss[maxn],F[maxn],G[maxn];
bool v[maxn];
void Prime(const int N){
	mu[1]=sig[1]=1;int m=0;
	for(int i=2;i<=N;i++){
		if(!v[i]) p[++m]=i,mu[i]=-1,sig[i]=mps[i]=i+1;
		for(int j=1,k;j<=m&&((k=i*p[j])<=N);j++){
			v[k]=1;
			if(i%p[j]==0) {sig[k]=sig[i]/mps[i]*(mps[k]=mps[i]*p[j]+1);break;}
			sig[k]=sig[i]*(mps[k]=p[j]+1),mu[k]=-mu[i];
		}
	}
	for(int i=1;i<=N;i++) ss[i]=(ss[i-1]+sig[i])%mod;
	for(int i=1;i<=N;i++)
		for(int j=1,k=i;k<=N;k+=i,j++) if(mu[j]){
			int tmp=1ll*mu[j]*j*j%mod*i%mod*sig[i]%mod;
			F[k]=(F[k]+1ll*tmp*ss[i])%mod,
			G[k]=(G[k]+1ll*tmp*sig[i])%mod;
		}
	for(int i=1;i<=N;i++) F[i]=(F[i]+F[i-1])%mod,G[i]=(G[i]+G[i-1])%mod;
}
int main()
{
	Prime(maxn-5);
	scanf("%d",&T);
	for(int t=1;t<=T;t++) scanf("%d",&n),printf("Case #%d: %d\n",t,(2ll*F[n]-G[n]+3ll*mod)%mod);
}