算法进阶——例题

移位运算例题

$求a的b次方对p取模的值,其中1\le a,b,p\le 10^9$

思路

根据数学常识，每个正整数可以唯一标识为若干指数不重复的$2$的次幂的和。
也就是说，如果$b$在二进制表示下有$k$位，其中第$i(0\le i<k)$位的数字时$c_i$，那么：
$$b=c_{k-1}{2}^{k-1}+c_{k-2}{2}^{k-2}+\cdots +c_0{2}^0$$
于是
$$a^b=a^{c_{k-1}\ast 2^{k-1}}+a^{c_{k-2}\ast 2^{k-2}}+\cdots +a^{c_0{2}^0}$$
因为$k=[lon{g}_2(b+1)]$(其中[ ]表示向上取整)，所以上市乘积项的数量不多于$[lo{g}_2(b+1)]$个。又因为
$$a^{2^i}=(a^{2^{i-1}}{)}^2$$
所以我们很容易通过$k$次递推求出每个乘积项，当$c_i=1$时,把该乘积项累积到答案中。$b\&1$运算可以取出$b$在二进制下的最低位，而$b>>1$运算可以舍去最低位，在递推的过程中将二者结合，就可以遍历$b$在二进制表示下的所有数位$c_i$。整个算法的时间复杂度为$O({\log }_{2}b)$

代码

	//calculate (a^b)modp
	int power(int a, int b, int p)
	{
		int ans = 1%p;
		for(;b;b>>=1)
		{
			if(b&1)
				ans = (long long)ans*a%p;
			a = (long long)a*a%p;
		}
		return ans;
	}

注意

若变量超出longlong的存储类型时，可以参考以下解决办法算法进阶——64位整数乘法