目录
快速幂
用于快速计算 (a^b) % p 这类式子
原理:
把b看作二进制数,从低位向高位遍历:
- 若当前位二进制数位0,则意味着还要进行偶数次幂运算,可以让 a = a*a以便后面直接一次运算
- 若当前位二进制数为1,则意味着还要进行奇数次幂运算,直接让 res = res * a 计算此次的结果
因为正整数的二进制最高有效位一定是1,所以最后计算出的res即为结果(记得边算边取余)
一个整数n的二进制位数为log(n),所以快速幂的复杂度为log(n)
long long fastPow(long long a,long long b,long long p)
{
long long ans = 1;
while(b)
{
if(b&1)
{
ans = (ans*a) % p;
}
a = (a*a) % p;
b = b>>1;
}
return ans;
}
龟速乘
对于(a * b) % p这类式子,计算过程很可能会超过long long的最大值
即使化为((a%p) * (b%p)) % p也会爆掉,例如:(1e17 * 1e17) % 1e18+3
此时我们需要用到龟速乘来模拟乘法边取余边乘来防止爆掉
原理:乘法具有分配律(a = x+y 则 a*b = (x*b) + (y*b)),最简单的方法就是对二进制下进行拆分
和快速幂原理大致相同,只是乘法换作了加法,把乘法看作加法来运算:
- 若当前位二进制数位0,则意味着还要进行偶数次加法,可以让 a = a+a 以便后面直接一次运算
- 若当前位二进制数为1,则意味着还要进行奇数次加法,直接让 res = res + a 计算此次的结果
计算的同时注意取模,保证每次的计算过程不会大于2p
龟速乘保证数据不溢出的代价是耗费了时间,时间复杂度由1变为了log(n),注意在可能溢出的时候再用
long long slow_mul(long long a,long long b,long long p)
{
long long res = 0;
while(b)
{
if(b&1)
{
res = (res + a) % p;
}
//a<<1与a*2与a+a相同
a = (a<<1) % p;
b = (b>>1);
}
return res;
}