51Nod1584 加权约数和-题解

又是一道非常神奇的数论题，学到了很多又花了一晚上QWQ

【题目地址】

题目大意

多组询问$T\le 50000$，每次给定一个$n\le 10^6$，询问下面式子在$\mathrm{m}\mathrm{o}\mathrm{d}10^9+7$意义下的值：

$$\sum _{i=1}^n\sum _{j=1}^n\max (i,j)\times {\sigma }_1(ij)$$

其中${\sigma }_1(i)$表示$i$的约数之和。

---

这次居然连$\max$都跑进去了，emmmm…

我们还是先把不好看的$\max$消除掉，所以对于每个$i$，我们枚举比它小的$j$，那么$\max (i,j)=i$，由于$(i,j),(j,i)$算两次，但是$(i,i)$算一次，所以原式转化为：

$$2\times \left(\sum _{i=1}^n\sum _{j=1}^{i}i\times {\sigma }_1(ij)\right)-\sum _{i=1}^{n}i{\sigma }_1(i^2)$$

我们把它分成两部分分析：

---

一个是前半部分：

$$\sum _{i=1}^n\sum _{j=1}^{i}i\times {\sigma }_1(ij)$$

根据这个题的经验，我们显然可以将后面的那一个${\sigma }_1(ij)$变换掉：

$$\begin{array}{rl}\sum _{i=1}^n\sum _{j=1}^{i}i\times {\sigma }_1(ij)=& \sum _{i=1}^n\sum _{j=1}^{i}i\sum _{x|i}\sum _{y|j}[gcd(x,y)=1]\frac{xj}{y}\end{array}$$

然后大都是这个题的套路，我们作如下变换：
$$\begin{array}{rl}\sum _{i=1}^n\sum _{j=1}^{i}i\times {\sigma }_1(ij)=& \sum _{i=1}^n\sum _{j=1}^{i}i\sum _{x|i}\sum _{y|j}[gcd(x,y)=1]\frac{xj}{y}\\ =& \sum _{i=1}^{n}i\sum _{j=1}^i\sum _{x|i}\sum _{y|j}\frac{xj}{y}\sum _{w|x,w|y}\mu (w)\\ =& \sum _{w=1}^n\mu (w)\sum _{w|x}x\sum _{x|i}i\sum _{w|y}\sum _{y|j}\frac{j}{y}\\ =& \sum _{w=1}^n\mu (w)\sum _{x=1}^{\lfloor \frac{n}{w}\rfloor }xw\sum _{x|i}iw\sum _{y=1}^{\lfloor \frac{n}{w}\rfloor }\sum _{y|j}\frac{jw}{yw}\\ =& \sum _{w=1}^n\mu (w)w^2\sum _{x=1}^{\lfloor \frac{n}{w}\rfloor }x\sum _{x|i}i\sum _{y=1}^{\lfloor \frac{n}{w}\rfloor }\sum _{y|j}\frac{j}{y}\\ =& \sum _{w=1}^n\mu (w)w^2\sum _{i=1}^{\lfloor \frac{n}{w}\rfloor }i\sum _{x|i}x\sum _{j=1}^{\lfloor \frac{n}{w}\rfloor }\sum _{y|j}\frac{j}{y}\end{array}$$

因为${\sum }_{y|j}\frac{j}{y}={\sum }_{y|j}y$，而根据在这个题目的技巧,我们知道${\sigma }_1(j)={\sum }_{y|j}y$，所以上面继续可以变换为：

$$\begin{array}{rl}\sum _{w=1}^n\mu (w)w^2\sum _{i=1}^{\lfloor \frac{n}{w}\rfloor }i\sum _{x|i}x\sum _{j=1}^{\lfloor \frac{n}{w}\rfloor }\sum _{y|j}\frac{j}{y}=& \sum _{w=1}^n\mu (w)w^2\sum _{i=1}^{\lfloor \frac{n}{w}\rfloor }i{\sigma }_1(i)\sum _{j=1}^{\lfloor \frac{n}{w}\rfloor }{\sigma }_1(j)\end{array}$$

然后我们可以$O(n)$的筛出${\sigma }_1,\mu$，并预处理上面的所用到的前缀和。

---

一个是后半部分：

$$\sum _{i=1}^{n}i{\sigma }_1(i^2)$$

这个处理就和上面的十分类似了：

$$\begin{array}{rl}\sum _{i=1}^{n}i{\sigma }_1(i^2)=& \sum _{i=1}^{n}i\sum _{x|i}\sum _{y|i}[gcd(x,y)=1]\frac{ix}{y}\\ =& \sum _{i=1}^{n}i\sum _{x|i}\sum _{y|i}\frac{ix}{y}\sum _{w|x,w|y}\mu (w)\\ =& \sum _{w=1}^n\mu (w)\sum _{w|i}i\sum _{w|x,x|i}x\sum _{w|y,y|i}\frac{i}{y}\\ =& \sum _{w=1}^n\mu (w)\sum _{i=1}^{\lfloor \frac{n}{w}\rfloor }iw\sum _{x|i}xw\sum _{y|i}\frac{iw}{yw}\\ =& \sum _{w=1}^n\mu (w)w^2\sum _{i=1}^{\lfloor \frac{n}{w}\rfloor }i\sum _{x|i}x\sum _{y|i}\frac{i}{y}\\ =& \sum _{w=1}^n\mu (w)w^2\sum _{i=1}^{\lfloor \frac{n}{w}\rfloor }i{\sigma }_1(i){\sigma }_1(i)\\ =& \sum _{w=1}^n\mu (w)w^2\sum _{i=1}^{\lfloor \frac{n}{w}\rfloor }i{\sigma }_1^2(i)\end{array}$$

---

此时，我们可以同样的预处理，然后带入原式，每次数论分块，然后$O(\sqrt{n})$的回答，复杂度大概$O(n+T\sqrt{n})$，但是最大数据有$10^6+2\times 50000\times 1000$，差不多一亿左右，而且加上大常数，最大数据大概跑3s左右，虽然丧心病狂地卡卡常就能过，但是肯定还有更优秀的做法。

---

我们将上面的两个式子带回原式，作如下变换：

$$\begin{array}{rl}ans(n)=& 2\times \left(\sum _{w=1}^n\mu (w)w^2\sum _{i=1}^{\lfloor \frac{n}{w}\rfloor }i{\sigma }_1(i)\sum _{j=1}^{\lfloor \frac{n}{w}\rfloor }{\sigma }_1(j)\right)-\left(\sum _{w=1}^n\mu (w)w^2\sum _{i=1}^{\lfloor \frac{n}{w}\rfloor }i{\sigma }_1^2(i)\right)\\ =& \sum _{w=1}^n\mu (w)w^2\left(\sum _{i=1}^{\lfloor \frac{n}{w}\rfloor }(2\cdot i{\sigma }_1(i)-i{\sigma }_1^2(i))\sum _{j=1}^{\lfloor \frac{n}{w}\rfloor }{\sigma }_1(j)\right)\\ =& \sum _{w=1}^n\mu (w)w^2\left(\sum _{i=1}^{\lfloor \frac{n}{w}\rfloor }i{\sigma }_1(i)\left(2\cdot \sum _{j=1}^{\lfloor \frac{n}{w}\rfloor }{\sigma }_1(j)-{\sigma }_1(i)\right)\right)\end{array}$$

如果我们能预处理处所有的$ans(n)$，然后每次$O(1)$的回答，那就会优秀的多，所以我们发现$ans(n)$不好直接求，所以考虑一个神奇的东西：差分

其实这个技巧在前面做过的题目中出现过：Imagine大佬-Orz，Imagine这个题目就用的差分做的。

我们来看，差分后如何计算：

$$\begin{array}{rl}cans(n)=& ans(n)-ans(n-1)\end{array}$$

直接算，发现并不好算QAQ，所以我们考虑对于一个单个的$n$，那么定义它的贡献就为：

$$=\sum _{d|n}\mu (d)d^2\left(\left(2\cdot \sum _{i=1}^{\lfloor \frac{n}{d}\rfloor }{\sigma }_1(i)-{\sigma }_1(\lfloor \frac{n}{d}\rfloor )\right)\cdot \lfloor \frac{n}{d}\rfloor \cdot {\sigma }_1(\lfloor \frac{n}{d}\rfloor )\right)$$

其实这里的意思，对于每一个$cans(n)$，对其进行贡献的有它的因子$d$，然后带入上面的式子可以得到这个。

而$ans(n)$，就是$cans(1\sim n)$所有的贡献，所以我们此时再求出$cans$的前缀和就是$ans$了。

也就是我们将一个前缀和中的一些项拆了出来，而两个前缀和之间差的就是$n$的贡献，所以差分的$cans$就等于这个。

而上面的那个$cans$的预处理，我们用类似埃氏筛，枚举因子倍数$O(nlogn)$的预处理即可。

所以总的复杂度变成了$O(nlogn+T)$，就不卡常啦！代码还是比较好写。

51Nod官网上还有一种不同的题解，方法差不多但是又有很多不同的技巧，有兴趣的可以去官网看。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int M=1e6+10,N=1e5+10,MAX=1e6+1;
const ll Mod=1e9+7;
ll prime[N],sigma[M],mu[M],c[M],cnt,ans[M];
bool vis[M];
void init(){
	mu[1]=sigma[1]=1;
	for(ll i=2;i<MAX;i++){
		if(!vis[i]){
			prime[++cnt]=i;
			mu[i]=-1;
			sigma[i]=1+i;
			c[i]=1+i;
		}
		for(ll j=1,v;j<=cnt&&i*prime[j]<MAX;j++){
			v=i*prime[j];
			vis[v]=1;
			if(!(i%prime[j])){
				c[v]=c[i]*prime[j]+1;
				sigma[v]=sigma[i]/c[i]*c[v];
				break;
			}
			mu[v]=-mu[i];
			sigma[v]=sigma[i]*sigma[prime[j]];
			c[v]=prime[j]+1;
		}
	}
	ll sum=1;
	for(ll i=2;i<MAX;i++){
		sum=(sum+sigma[i])%Mod;
		sigma[i]=(i*sigma[i]%Mod*((2ll*sum-sigma[i])%Mod)%Mod)%Mod;
		if(sigma[i]<0)sigma[i]+=Mod;
	}
	for(ll i=1,v;i<MAX;i++){
		v=((i*i)%Mod*mu[i])%Mod;
		if(v<0)v+=Mod;
		for(ll j=1;i*j<MAX;j++){
			ans[i*j]=(ans[i*j]+v*sigma[j]%Mod)%Mod;
		}
	}
	for(ll i=2;i<MAX;i++)ans[i]=(ans[i-1]+ans[i])%Mod;
}
int T,n;
int main(){
	init();
	scanf("%d",&T);
	for(int cas=1;cas<=T;cas++){
		scanf("%d",&n);
		printf("Case #%d: %lld\n",cas,ans[n]);
	}
	return 0;
}