HDU - 7318 Guess（ 2023“钉耙编程”中国大学生算法设计超级联赛第四场 G）

题目大意

最近，$\text{Stump}$ 推导出了 ${\sum }_{k=1}^n{\mu }^2(k)={\sum }_{k=1}\mu (k)\lfloor \frac{n}{k^2}\rfloor$，这震惊了 $\text{Yoshinow2001}$ 一整年。

上面的 $\mu$ 是莫比乌斯函数：如果 $n$ 包含平方因子（即有正整数 $a>1$ 使 $a^2|n$），那么 $\mu (n)=0$。否则，不妨分解质因数 $n=p_1{p}_2\cdots p_k$，然后 $\mu (n)=(-1{)}^k$。例如：$\mu (1)=1.\mu (2)=\mu (3)=-1$。

回想一下 $\ln (n)$ 表示以 $e$ 为底的 $n$ 的对数，其中 $e={\sum }_{i=1}^{\infty }\frac{1}{i!}\approx 2.71828$。

现在 $\text{Yoshinow2001}$ 非常愤怒，提出了一个问题！

让：

$$S(n)=\sum _{d|n}\mu (\frac{n}{d})\ln (d)$$

你需要计算:

$$e^{S(n)}\mathrm{mod}998244353$$

当 $\text{Stump}$ 看到题目时，他气坏了!他没有推出 $O(1)$ 的判断规律。

现在你要告诉他什么叫做大佬，写出一个程序在 $5s$ 内爆切这个问题吧！

思路

我们先把 $S(n)$ 代入进去，我们要求的式子变为如下形式：

$$e^{{\sum }_{d|n}\mu (\frac{n}{d})\ln (d)}$$

我们知道 $e^{ln(n)}=n$，则我们可以将式子化为如下形式：

$$\prod _{d|n}(\frac{n}{d}{)}^{\mu (d)}=\frac{{\prod }_{d|n}{n}^{\mu (d)}}{{\prod }_{d|n}{d}^{\mu (d)}}$$

化为两个式子后我们分别来看，先看上面的式子。

由于底数不变，则式子可以写成如下形式：

$$n^{{\sum }_{d|n}\mu (d)}$$

因为 ${\sum }_{d|n}\mu (d)=0$（莫比乌斯函数的性质），所以这个式子的值就为 $1$。

再来看下面的式子。

我们对于 $n$ 质因数分解，得 $n=p_1^{k_1}{p}_2^{k_2}\cdots p_m^{k_m}$。

由于莫比乌斯函数的定义，所以 $d$ 不可能包含重复的质因数。

于是我们考虑每个质因数的贡献，可以列出如下式子。

$$\prod _{i=1}^m{p}_i^{{\sum }_{j=0}^{k-1}{C}_{k-1}^j(-1{)}^{j-1}}$$

这一步本质上就是将 $d$ 质因数分解，由于连乘可以将指数相加，得出以上式子。

则原式变为：

$$\frac{1}{{\prod }_{i=1}^m{p}_i^{{\sum }_{j=0}^{k-1}{C}_{k-1}^j(-1{)}^{j-1}}}=\prod _{i=1}^m{p}_i^{{\sum }_{j=0}^{k-1}{C}_{k-1}^j(-1{)}^j}$$

我们考虑 ${\sum }_{j=0}^{k-1}{C}_{k-1}^j(-1{)}^j$，这个式子只有当 $k-1>0$，即 $k>1$ 的时候才为 $1$，其余时候皆为 $0$。

所以只有当 $n$ 只有一种质因子时，答案为 $n$ 的质因子，否则答案为 $1$。

我们可以枚举 $n$ 开 $k$ 次方后的数，在判断这个数 $k$ 次方等不等于 $n$，再用 $\text{miller\_rabin}$ 来判断是否为质数，如果为质数，则输出该质数。

如果找不到满足条件的质数，则输出 $1$。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
int T, bz[1000005], pn;
LL n, m, ans, p[500000];
__int128 Pow(__int128 a, __int128 b, __int128 MOD) {
    __int128 s = 1;
    while (b) {
        if (b & 1)
            s = s * a % MOD;
        a = a * a % MOD, b = b >> 1;
    }
    return s;
}
bool miller_rabin(LL x) {
    if (x < 2)
        return false;
    for (int i = 1; i <= pn; i++) {
        if (p[i] == x)
            return true;
        if (x % p[i] == 0)
            return false;
        if (Pow(p[i], x - 1, x) != 1)
            return false;
        LL k = x - 1, t;
        while (k % 2 == 0) {
            k = k / 2, t = Pow(p[i], k, x);
            if (t != 1 && t != x - 1)
                return false;
            if (t == x - 1)
                return true;
        }
    }
    return true;
}
int main() {
    for (int i = 2; i <= 1000000; i++) {
        if (!bz[i])
            p[++pn] = i;
        for (int j = 2; j * i <= 1000000; j++) bz[j] = 1;
    }
    scanf("%d", &T);
    while (T--) {
        scanf("%lld", &n), ans = n;
        if (n == 1) {
            printf("1 ");
            continue;
        }
        bool flag = false;
        for (int i = 1; i <= 64; i++) {//注意精度问题，我们可以将开i次方后的数+1或-1再尝试一下
            LL t = pow((long double)n, (long double)1.0 / i);
            if (miller_rabin(t) && Pow(t, i, n + 1) == n) {
                flag = true, printf("%lld ", t % 998244353);
                break;
            }
            t = pow((long double)n, (long double)1.0 / i) + 1;
            if (miller_rabin(t) && Pow(t, i, n + 1) == n) {
                flag = true, printf("%lld ", t % 998244353);
                break;
            }
            t = pow((long double)n, (long double)1.0 / i) - 1;
            if (miller_rabin(t) && Pow(t, i, n + 1) == n) {
                flag = true, printf("%lld ", t % 998244353);
                break;
            }
        }
        if (!flag)
            printf("1 ");
    }
}