bzoj3309 DZY Loves Math

题面：bzoj3309

题意：设$f(n)$表示$n$所含质因子的最大幂指数。求${\sum }_{i=1}^a{\sum }_{j=1}^{b}f(gcd(i,j))$

题解：

$\begin{array}{rl}ans& =\sum _{i=1}^a\sum _{j=1}^{b}f(gcd(i,j))\\ & =\sum _{d=1}^{a}f(d)\sum _{i=1}^a\sum _{j=1}^b[gcd(i,j)=d]\\ & =\sum _{d=1}^{a}f(d)\sum _{i=1}^{\frac{a}{d}}\mu (i)\lfloor \frac{a}{id}\rfloor \lfloor \frac{b}{id}\rfloor \end{array}$

令$T=id$，$ans={\sum }_{T=1}^a\lfloor \frac{a}{T}\rfloor \lfloor \frac{b}{T}\rfloor {\sum }_{d\mid T}f(d)\mu (\frac{T}{d})$

关键在于如何求${\sum }_{d\mid T}f(d)\mu (\frac{T}{d})$

考虑$T=p_1^{a_1}{p}_2^{a_2}\dots p_n^{a_n}$

$\mu (\frac{T}{d})$非零采用贡献，因此$\frac{T}{d}$的每个质因数最多取1，总共有$2^n$个$d$

假设选定某个质因数$p_x^{a_x-1}$作为最高次幂，那么其他比它低的幂次都可以选或者不选，共有$2^k$个。

$f(d)$值不变，$\mu (\frac{T}{d})$一一对应-1、1，因此${\sum }_{d\mid T}f(d)\mu (\frac{T}{d})=0$

因此如果幂次不全部相等，和为0。

假设幂次全部相等，即$T=(p_1{p}_2\dots p_n{)}^a$

$d=(p_1{p}_2\dots p_n{)}^{a-1}$时，$f(d)=a-1$

其他情况下$f(d)=a$

先假设所有$f(d)=a$，显然${\sum }_{d\mid T}\mu (\frac{T}{d})=0$

但是因为有一种情况$f(d)=a-1$，因此要减去$f(d)$中的1

贡献是$-1\ast \mu (p_1{p}_2{p}_3\dots p_n)=(-1{)}^{n+1}$

线性筛时记录一下去掉最小质因子后的数$lst[i]$，最小质因子的次数，以及${\sum }_{d\mid T}f(d)\mu (\frac{T}{d})$

#include <bits/stdc++.h>
using namespace std;
typedef long long ll ;
const int maxn = 1e7 + 10 ;
int prime[maxn], tot ;
int lst[maxn], fp[maxn], g[maxn] ;
bool vis[maxn] ;
void init () {
    for (int i = 2; i < maxn; i ++) {
        if (!vis[i]) {
            prime[++ tot] = i ;
            lst[i] = fp[i] = g[i] = 1 ;
        }
        for (int j = 1; j <= tot && i * prime[j] < maxn; j ++) {
            int x = i * prime[j] ;
            vis[x] = 1 ;
            if (i % prime[j] == 0) {
                lst[x] = lst[i] ;
                fp[x] = fp[i] + 1 ;
                if (lst[x] == 1) g[x] = 1 ;
                else g[x] = (fp[x] == fp[lst[x]] ? -g[lst[x]] : 0) ;
                break ;
            }
            lst[x] = i; fp[x] = 1; g[x] = (fp[i] == 1 ? -g[i] : 0) ;
        }
    }
    for (int i = 1; i < maxn; i ++) g[i] += g[i - 1] ;
}
int main() {
    init () ;
    int T ;
    cin >> T ;
    while (T --) {
        int a, b ;
        scanf("%d%d", &a, &b) ;
        if (a > b) swap (a, b) ;
        ll ans = 0 ;
        for (int i = 1, j; i <= a; i = j + 1) {
            j = min (a / (a / i), b / (b / i)) ;
            ans += 1ll * (a / i) * (b / i) * (g[j] - g[i - 1]) ;
        }
        printf("%lld\n", ans) ;
    }
    return 0 ;
}