快速幂(超级好的板子)

2023-03-09 16:03:46 浏览数 (1)

    快速幂就是快速算底数的n次幂。其时间复杂度为 O(log₂N), 与朴素的O(N)相比效率有了极大的提高。

    代码实现:

代码语言:javascript复制
int pow_mod(int a, int k)
{
    int ans = 1;
    while(k)
    {
        if(k % 2)
            ans *= a;
        a *= a;
        k /= 2;
    }
    return ans;
}

        原理我理解的就是把指数换成二进制进行运算,如果二进制位数的数是1,那么让我们要的结果ans乘底数a,然后a要自乘一次,如果是二进制位数的数是0,那么ans不进行操作,a还是要自乘一次的,k就是指数,比如2^6,这里a就是2,k就是6。

       对于快速幂取模可以利用a * b % m = [(a % m) * (b % m)] % m;

        代码实现:

代码语言:javascript复制
long long int pow_mod(long long int a, long long int k)
{
    long long int ans = 1;
    a %= MOD;
    while(k)
    {
        if(k % 2) ans *= a;
        a = (a * a) % MOD;
        k /= 2;
        ans %= MOD;
    }
    return ans;
}

0 人点赞