算法进阶——64位整数乘法

移位运算例题

$求a乘b对p取模的值，其中1\le a,b,p\le 10^{18}$

方法一

类似于快速幂的思想，把整数b用二进制表示,即
$$b=c_{k-1}\ast 2^{k-1}+c_{k-2}\ast 2^{k-2}+\cdots +c_0\ast 2^0$$
那么
$$a\ast b=c_{k-1}\ast a\ast 2^{k-1}+c_{k-2}\ast a\ast 2^{k-2}+\cdots +c_0\ast a\ast 2^0$$
因为
$$a\ast 2^i=(a\ast 2^{i-1})\ast 2$$

若已求出$a\ast 2^{i-1}\mathrm{mod}p$ ，则计算 $ (a*2^{i-1})*2 \mod p$ 时 ，运算过程中的每一步都不超过$2\ast 10^{1}8$，在64位整数 longlong的表示范围内，很容易通过k次递推求出每个乘机项。当$c_i=1$时，把该成绩项累加到答案中即可，时间复杂度为$O(lo{g}_{2}b)$

代码

long long mul(long long a, long long b, long long p)
{
	long long ans = 0;
	for(;b;b>>=1)
	{
    	if(b&1)
      		ans = (ans+1)%p;
    	a = a*2%p;
  	}
  	return ans;
}

方法二

利用$a\ast b\mathrm{mod}p=a\ast b-[a\ast b/p]\ast p$，其中[ ]表示下取整

首先，当$a,b<p$时，$a\ast b/p$下取整后也一定小于p。我们可以利用浮点数执行$a\ast b/p$的运算，而不用关心小数点之后的部分。浮点类型 long double 在十进制下有效数字有18 到19位，足够胜任。当浮点数精度不足以保存精确数值时，它会像科学计数法一样舍弃低位，正好符合需求

另外虽然$a\ast b$和$[a\ast b/p]\ast p$的可能很大，但是二者的差一定在0到$p-1$之间，我们只关心他们较低的几位即可。因此，我们可以用 long long 来保存$a\ast b$和$[a\ast b/p]\ast p$各自的结果。整数运算一处相当于舍弃高位，也符合需求

代码

	long long mul(long long a, long long b, long long p)
	{
		a %= p , b%=p;//当a,b一定在0~p之间时，此行不必要
		long long c = (long double)a*b/p;
		long long ans = a*b-c*p;
		if(ans<0)
			ans += p;
		else if(ans>=p)
			ans -= p;
		return ans;
	}