运用O(1)快速乘的快速幂写法以及注意事项

JSU_曾是此间年少

于 2023-03-13 10:04:22 发布

阅读量163

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/m0_72678953/article/details/129485970

版权

文章讨论了如何使用快速幂算法解决大整数幂运算，并提出了进一步的优化策略，即通过快速乘方法提升计算速度。在原基础上，当执行`res*a%mod`时，改用O(1)的快速乘公式`(a*b-((ll)((longdouble)a/mod*b))*mod+mod)%mod`，同时强调了在计算前对a和b取模的重要性。最后展示了优化后的快速幂算法实现。

摘要由CSDN通过智能技术生成

题目，给定较大的a,b,c 求a^b%c

首先看到题目是用基础的快速幂算法，其模板如下：

inline ll ksm(ll a,ll b,ll mod)
{
    if (mod==1)return 0;
    ll res=1;
    while(b)
    {
        if(b&1)
            res=res*a%mod;
        b>>=1;
        a=a*a%mod;
    }
    return res;
}

这就已经很快了吗？还可以继续优化！

比如看到res*a%mod

可以使用O(1)快速乘来继续优化速度，其模板如下

ll mul(ll a,ll b,ll mod)
{
    a %= mod;
    b %= mod;
    return (a*b-((ll)((long double)a/mod*b))*mod+mod)%mod;
}

注意事项：

1. a,b 使用前先取模，不然可能会被卡样例

2. long double需要将a先转成 long double 型随后对其代数式整体转LL

原始的快速乘

inline ll ksc(ll a, ll b, ll c)
{
    ll ans = 0;
    while(b)
    {
        if(b & 1)ans = (ans + a) % c;
        b >>= 1;
        a = (a + a) % c;
    }
    return ans;
}

结合以上代码，我们写出优化后的快速幂

ll mul(ll a,ll b,ll mod)
{
    a %= mod;
    b %= mod;
    return (a*b-((ll)((long double)a/mod*b))*mod+mod)%mod;
}

inline ll ksm(ll a,ll b,ll mod)
{
    if (mod==1)return 0;
    ll res=1;
    while(b)
    {
        if(b&1)
            res=mul(res,a,mod);
        b>>=1;
        a=mul(a,a,mod);
    }
    return res;
}