快速幂核心
a^b mod c=(a^2)^(b/2) mod c (b为偶数);
a^b mod c=((a^2)^(b div 2)*a) mod c (b为奇数)
以一道题为例,讲解快速幂算法。
题目:计算a^b mod c
朴素算法:直接计算求值。
var a,b,c,i,ans:longint;
begin
readln(a,b,c);
ans:=1;
fori:=1 to b do
ans:=ans*a;
ans:=ans mod c;
writeln(ans);
end.
以上算法有一个缺点:当a,b很大时,容易超出longint范围。
因为a^b mod c=(a mod c)^b mod c
得出改进型算法:在计算过程中不断进行mod运算
var a,b,c,i,ans:longint;
begin
readln(a,b,c);
a:=a mod c;
ans:=1;
fori:=1 to b do
ans:=(ans*a) mod c;
ans:=ans mod c;
writeln(ans);
end.
时间复杂度:O(n)
这种算法也有一定缺陷:当c很大时,容易超时。
因为a^b mod c=(a^2)^(b/2) mod c (b为偶数);
a^b mod c=((a^2)^(b div 2)*a) mod c (b为奇数)
所以可以在计算过程中不断地将底数平方、指数除以2,用另一个变量记录最终答案。
得到快速幂算法:
var a,b,c,ans:int64;
begin
readln(a,b,c);
write(a,'^',b,' mod ',c,'=');
a:=a mod c;
ans:=1;
while b>0 do
begin
if bmod 2=1 then ans:=ans*a mod c;
b:=bdiv 2;
a:=(a*a)mod c;
end;
writeln(ans);
end.
时间复杂度:O(log2n),能在竞赛通过几乎全部的数据,是目前最常用的算法之一。
import java.math.BigInteger;
import java.util.Scanner;
public class Main {
public static BigInteger result(BigInteger x,BigInteger y,BigInteger z)//快速幂
{
BigInteger sum=BigInteger.ONE;
while(y.compareTo(BigInteger.ZERO)==1)
{
if(y.mod(BigInteger.valueOf(2)).compareTo(BigInteger.ONE)==0)
{
sum=sum.multiply(x);
sum=sum.mod(z);
}
y=y.divide(BigInteger.valueOf(2));
x=x.multiply(x);
x=x.mod(z);
}
return sum;
}
public static void main(String[] args) {
Scanner scanner=new Scanner(System.in);
int cases=scanner.nextInt();
while(cases--!=0)
{
BigInteger a,b,c;
a=scanner.nextBigInteger();
b=scanner.nextBigInteger();
c=scanner.nextBigInteger();
System.out.println(result(a,b,c));
}
}
}