【LOJ 125】除数函数求和 2

【题目】

传送门

题目描述：

求 $\sum _{i=1}^{n}2{\sigma }_2(i)+3{\sigma }_1(i)+5{\sigma }_0(i)$ 对 $998244353$ 取模。

其中，${\sigma }_k(i)=\sum _{d|i}{d}^k$。

输入格式：

一行一个正整数 $n$。

输出格式：

一行一个正整数表示答案对 $998244353$ 取模的结果。

样例数据：

【样例 $1$】

输入
2

输出
39

【样例 $2$】

输入
10

输出
1334

【样例 $3$】

输入
1000000

输出
620771835

数据范围与提示：

对于 $100\%$ 的数据，$n\le {10}^9$。

【分析】

这道题的思路其实和Calculating差不多。

就以 ${\sigma }_2(k)$ 举例吧，${\sigma }_1(k)$ 和 ${\sigma }_2(k)$ 本质上是一样的。

对于 $i$，在 $n$ 以内它有 $\lfloor \frac{n}{i}\rfloor$ 个倍数，那 $i$ 产生的贡献就是 $i^2\lfloor \frac{n}{i}\rfloor$。

因此，$\sum _{i=1}^n{\sigma }_2(i)=i^2\lfloor \frac{n}{i}\rfloor$。

同理，$\sum _{i=1}^n{\sigma }_1(i)=i\lfloor \frac{n}{i}\rfloor$，$\sum _{i=1}^n{\sigma }_0(i)=\lfloor \frac{n}{i}\rfloor$。

那原式就可以化简了

$$\sum _{i=1}^n\lfloor \frac{n}{i}\rfloor (2i^2+3i+5)$$

进一步化简就是

$$2\sum _{i=1}^n{i}^2\lfloor \frac{n}{i}\rfloor +3\sum _{i=1}^{n}i\lfloor \frac{n}{i}\rfloor +5\sum _{i=1}^n\lfloor \frac{n}{i}\rfloor$$

用整除分块就可以轻松解决。

这里补一个小知识，即如何计算 $\sum _{i=1}^n{i}^2$。（整除分块计算时要用到）
$\because (n+1{)}^3=n^3+3n^2+3n+1$
$\therefore \sum _{i=2}^{n+1}{i}^3=\sum _{i=1}^n{i}^3+3\sum _{i=1}^n{i}^2+3\sum _{i=1}^{n}i+n$
$\therefore 3\sum _{i=1}^n{i}^2=(n+1{)}^3-1^3-3\frac{(1+n)\cdot n}{2}-n$
$\therefore \sum _{i=1}^n{i}^2=\frac{n(n+1)(2n+1)}{6}$

【代码】

#include<cstdio>
#include<cstring>
#include<algorithm>
#define Mod 998244353
#define inv2 499122177
#define inv6 166374059
using namespace std;
long long calc(int n)
{
	return 1ll*n*(n+1)%Mod*(2*n+1)%Mod*inv6%Mod;
}
int main()
{
	int n,i,j;
	scanf("%d",&n);
	long long s1=0,s2=0,s3=0;
	for(i=1;i<=n;i=j+1)
	{
		j=n/(n/i);
		s1=(s1+1ll*(calc(j)-calc(i-1)+Mod)%Mod*(n/i)%Mod)%Mod;
		s2=(s2+1ll*(i+j)*(j-i+1)%Mod*inv2%Mod*(n/i)%Mod)%Mod;
		s3=(s3+1ll*(j-i+1)*(n/i))%Mod;
	}
	printf("%lld",(2*s1+3*s2+5*s3)%Mod);
	return 0;
}