POJ 1845 逆元 / 分治

SHOHOKUKU

于 2020-11-17 00:01:20 发布

阅读量165

点赞数

分类专栏：数学基本算法

本文链接：https://blog.csdn.net/neweryyy/article/details/109732499

版权

基本算法同时被 2 个专栏收录

252 篇文章 1 订阅

订阅专栏

数学

164 篇文章 0 订阅

订阅专栏

题意

传送门 POJ 1845

题解

分解质因数 $A=p_1^{e_1}p_2^{e_2}\dots p_n^{e_n}$ ，则约数和为
$(1+p_1+\dots +p_1^{e_1})(1+p_2+\dots +p_2^{e_2})\dots (1+p_n+\dots +p_n^{e_n})$

逆元

对于质因数 $p_i$ ，根据等比数列求和公式，括号内求和为 $\frac{p_i^{e_iB+1}-1}{p_i-1}$ 分子项快速幂求解，分母项逆元求解。对于 $p_i\ mod\ m =1$ 的情况需要特殊处理，因为此时 $p_i-1$ 不存在逆元。对于正整数 $N$ ，至多有一个大于等于 $\sqrt N$ 的质因子，那么逆元打表处理 $[1,\sqrt N]$ 的部分，对于超出这个范围的部分，利用费马小定理进行快速幂求逆元。

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <map>
using namespace std;
#define mod 9901
typedef map<int, int> mp;
int A, B, inv[mod];

mp prime_factor(int n)
{
    mp res;
    for (int i = 2; i * i <= n; ++i)
    {
        while (n % i == 0)
        {
            ++res[i];
            n /= i;
        }
    }
    if (n != 1)
        res[n] = 1;
    return res;
}

int mod_pow(int x, int n)
{
    int res = 1;
    x %= mod;
    while (n)
    {
        if (n & 1)
            res = res * x % mod;
        x = x * x % mod;
        n >>= 1;
    }
    return res;
}

int get_inv(int n)
{
    if (n < mod)
        return inv[n];
    return mod_pow(n, mod - 2);
}

int main()
{
    inv[1] = 1;
    for (int i = 2; i < mod; ++i)
        inv[i] = (mod - mod / i) * inv[mod % i] % mod;
    scanf("%d%d", &A, &B);
    mp fac = prime_factor(A);
    int res = 1;
    for (mp::iterator it = fac.begin(); it != fac.end(); ++it)
    {
        int p = it->first, e = it->second * B;
        if (p % mod != 1)
            res = res * (mod_pow(p, e + 1) + mod - 1) % mod * get_inv(p - 1) % mod;
        else
            res = res * (e % mod + 1) % mod;
    }
    printf("%d\n", res);
    return 0;
}

分治

考虑质因数 $p_i$ ，若 $e_i$ 为奇数，设其括号内求和为 $s u m (p, e)$ ，则有
$sum(p,e)=(1+p+\dots+p^{\lfloor e/2\rfloor})+(p^{\lfloor e/2\rfloor+1}+\dots +p^e)=(1+p^{\lfloor e/2\rfloor+1})\times sum(p,\lfloor e/2\rfloor)$ 指数为偶数，将 $p_i^{e_i}$ 加入答案后将 $e_i$ 减一处理为奇数情况即可。

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <map>
using namespace std;
#define mod 9901
typedef map<int, int> mp;
int A, B;

mp prime_factor(int n)
{
    mp res;
    for (int i = 2; i * i <= n; ++i)
    {
        while (n % i == 0)
        {
            ++res[i];
            n /= i;
        }
    }
    if (n != 1)
        res[n] = 1;
    return res;
}

int mod_pow(int x, int n)
{
    int res = 1;
    x %= mod;
    while (n)
    {
        if (n & 1)
            res = res * x % mod;
        x = x * x % mod;
        n >>= 1;
    }
    return res;
}

int sum(int p, int e)
{
    if (!e)
        return 1;
    int res = 0;
    if (!(e & 1))
        res += mod_pow(p, e), --e;
    e /= 2, res += (1 + mod_pow(p, e + 1)) * sum(p, e);
    return res % mod;
}

int main()
{
    scanf("%d%d", &A, &B);
    mp fac = prime_factor(A);
    int res = 1;
    for (mp::iterator it = fac.begin(); it != fac.end(); ++it)
    {
        int p = it->first, e = it->second * B;
        res = res * sum(p, e) % mod;
    }
    printf("%d\n", res);
    return 0;
}