一本通1633：【例 3】Sumdiv

原创于 2024-04-14 08:38:31 发布 · 148 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #c++

本文详细介绍了如何使用C++编写一个函数qmi，结合扩展欧几里得算法来计算两个数在给定模数下的逆元。主函数中展示了如何利用这个函数解决模逆问题，最终输出计算结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <cstdio>
#define LL long long

const int mod = 9901;

LL qmi(LL a, LL b) {
    LL res = 1, mul = a;

    while (b) {
        if (b & 1)
            res = res * mul % mod;

        mul = mul * mul % mod;
        b >>= 1;
    }

    return res;
}

int main() {
    LL a, b, ans = 1;
    scanf("%lld %lld", &a, &b);

    for (int i = 2, j; i <= a / i; i ++)
        if (a % i == 0) {
            j = 0;

            while (a % i == 0)
                a /= i, j ++;

            j *= b;
            ans *= (1 - qmi(i, j + 1)) * qmi(1 - i, mod - 2) % mod;
            ans %= mod;
        }

    if (a > 1)
        ans *= (1 - qmi(a, b + 1)) * qmi(1 - a, mod - 2) % mod;

    ans %= mod;
    printf("%lld", ans);
    return 0;
}