快速幂
实现原理
假设我们要求a^b,按照朴素算法就是把a连乘b次,这样一来时间复杂度是O(b),即是O(n)级别。但快速幂能做到O(logn)的复杂度。
假设我们要求a^b,那么其实b是可以拆成二进制的,该二进制数第i位的权为 2 ^(i-1),例如当b==11时:
1.传统方法:11个a相乘
2.快速幂:
11的二进制是1011,
11 = 2³×1 + 2²×0 + 2¹×1 + 2º×1
a11=a^(20+21+23 )
a11=a20 * a21 * a23
a11=a1 * a2 * a8
原来算11次,现在算3次。
由于是二进制,很自然地想到用位运算这个强大的工具:&和>> 。
&运算通常用于二进制取位操作,例如一个数 & 1 的结果就是取二进制的最末位。还可以判断奇偶x&1=0为偶,x&1=1为奇。
>> 运算比较单纯,二进制去掉最后一位
其中要理解base*=base这一步:因为 base * base= =base2,下一步再乘,就是base2 * base2==base4,然后同理 base4 * base4=base8,由此可以做到base–>base2–>base4–>base8–>base16–>base32…指数正是 2^i ,再看上面的例子,a¹¹= a1 * a2 * a8,这三项就可以完美解决了,快速幂就是这样。
代码实现
public class PowTest{
public static void main(String[] args) {
int res = pow(2, 5);
System.out.println(res);
}
public static int pow(int a, int b) {
int ans = 1;
int base = a;
while (b != 0) {
if ((b & 1) != 0) {// 最后一位进行位运算,如果b的二进制最后为1,则结果为1
ans = ans * base;
}
base = base * base; // 每次得到a^2^n
b = b >> 1;
}
return ans;
}
}
快速幂取模
实现原理
快速幂实现的最基本的理论就是我们离散课上或者数论中学过的一条公式推出的引理。
引理:积的取余等于取余的积的取余。
(a * b) mod c = ((a mod c)*(b mod c)) mod c
再在这条引理的基础之上,对指数型数据进行拆分以及合并,从而得到我们用的快速幂算法。
(a^b) mod c = (a mod c)^b mod c
我们可以在求快速幂的时候,通过对底数取模的方式,不断缩小底数的规模。
代码实现
public class Pow_mod{
public static void main(String[] args) {
System.out.println(pow_mod(2, 5, 3));
}
public static int pow_mod(int a, int b, int c) {
// a是底数,b是指数,c是取模数
int ans = 1;
int base = a % c;
while (b != 0) {
if ((b & 1) != 0)
ans = (ans * base) % c;
base = (base * base) % c;// 不断的两两合并再取模,减小a和b的规模
b >>= 1;
}
return ans;
}
}