[51Nod 1584] 加权约数和
Description
在整理以前的试题时,他发现了这样一道题目:“求 \(\sum\sigma(i)\),其中 \(1≤i≤N\),\(σ(i)\) 表示 \(i\) 的约数之和。”
现在他长大了,题目也变难了,所以麻烦你来帮他解决一道数论题吧。
他需要你求如下表达式的值:
其中 \(\max(i,j)\) 表示 \(i\) 和 \(j\) 里的最大值,\(\sigma(i⋅j)\) 表示 \(i⋅j\) 的约数之和。
例如当 \(N=2\) 的时候,由 \(σ(1)=1,σ(2)=1+2=3,σ(4)=1+2+4=7\) 可知,答案应为 \(1⋅σ(1⋅1)+2⋅σ(1⋅2)+2⋅σ(2⋅1)+2⋅σ(2⋅2)=27\)。
他发现答案有点大,所以你只需要告诉他答案模 \(1000000007\) 的值即可。
Input
每个测试点含有多组测试数据。
第一行是一个正整数 \(T\),表示接下来有 \(T\) 组测试数据。
接下来的 \(T\) 行,每组测试数据占一行。
每行有一个正整数 \(N\),含义如描述所示。
Output
共有 \(T\) 行。对于每组测试数据,输出一行信息"Case #x: y"。
其中 \(x\) 表示对应的是第几组测试数据,\(y\) 表示相应的答案模 \(1000000007\) 的值。
Sample Input
5
1
2
3
4
5
Sample Output
Case #1: 1
Case #2: 27
Case #3: 162
Case #4: 686
Case #5: 1741
HINT
\(1≤T≤50000,1≤N≤1000000\)
Solution
〖一〗
证明:原式 \(=\sum\limits_{p\mid i}\sum\limits_{q\mid j}\left[\left(p,\dfrac{j}{q}\right)=1\right]p\cdot q\),设 \(i=\sum p_i^{a_i},j=\sum p_i^{b_i},p=\sum p_i^{c_i},q=\sum p_i^{d_i}\)。
- 若 \(0<c_i\le a_i\),则 \(d_i=b_i\),此时可以表示出 \(p_i^{(b_i+1)\rightarrow(b_i+a_i)}\);
- 若 \(c_i=0\),则 \(0\le d_i\le b_i\),此时可以表示出 \(p_i^{0\rightarrow b_i}\)。
综上,该式一定可以表示出 \(p_i^{0\rightarrow(b_i+a_i)}\)。
〖二〗
而这样只能做到 \(O(\sqrt n)\) 询问,于是进一步化简:
第一个式子中的 \(d\) 是 \(i\cdot d\) 的约数,第二个式子中的 \(i\) 表示 \(i\cdot d\),\(d\) 还是约数,因此两个式子相等。
然后就可以 \(O(n\ln n)\) 预处理,\(O(1)\) 查询了。
Code
#include <cstdio>
const int N = 1000005, mod = 1000000007;
int mu[N], np[N], p[N], tot, a[N], b[N], f[N], g[N], n = 1000000;
int read() {
int x = 0; char c = getchar();
while (c < '0' || c > '9') c = getchar();
while (c >= '0' && c <= '9') x = (x << 3) + (x << 1) + (c ^ 48), c = getchar();
return x;
}
void euler() {
mu[1] = 1;
for (int i = 2; i <= n; ++i) {
if (!np[i]) p[++tot] = i, mu[i] = -1;
for (int j = 1; j <= tot && i * p[j] <= n; ++j) {
np[i * p[j]] = 1;
if (i % p[j] == 0) { mu[i * p[j]] = 0; break; }
mu[i * p[j]] = -mu[i];
}
}
}
int main() {
euler();
for (int i = 1; i <= n; ++i)
for (int j = i; j <= n; j += i)
if ((a[j] += i) >= mod) a[j] -= mod;
for (int i = 1; i <= n; ++i) if ((b[i] = b[i - 1] + a[i]) >= mod) b[i] -= mod;
for (int i = 1; i <= n; ++i)
for (int j = i; j <= n; j += i) {
f[j] = (f[j] + 1LL * mu[i] * i * j % mod * a[j / i] % mod * b[j / i]) % mod;
g[j] = (g[j] + 1LL * mu[i] * i * j % mod * a[j / i] % mod * a[j / i]) % mod;
}
for (int i = 2; i <= n; ++i) {
if ((f[i] += f[i - 1]) >= mod) f[i] -= mod;
if (f[i] < 0) f[i] += mod; //mu可能是负数, 需要判断正负
if ((g[i] += g[i - 1]) >= mod) g[i] -= mod;
if (g[i] < 0) g[i] += mod;
}
int T = read();
for (int i = 1; i <= T; ++i) n = read(), printf("Case #%d: %lld\n", i, (2LL * f[n] + mod - g[n]) % mod);
return 0;
}