NuptOJ1037模运算——快速幂取模

模运算

时间限制(普通/Java):1000MS/3000MS          运行内存限制:65536KByte
总提交:759            测试通过:139

描述

给定整数a,b,n,要求计算(a^b)mod n

输入

多组数据,每组数据一行,为三个用空格隔开的整数a,b,n

1<=a<=5000,0<=b<=10^8,1<=n<=5000000

输出

每组数据输出一行,为所求值

样例输入

2 3 5
2 2 4

样例输出

3
0

题目来源

NUAA

分析：快速幂取模 算法。

#include<iostream>
using namespace std;

//模运算
//快速幂取模

int main()
{
	__int64 a, b, n;
	while(scanf("%I64d%I64d%I64d",&a,&b,&n) != EOF)
	{
		int ans = 1;
		a = a % n;
		while(b > 0)
		{
			if(b % 2 == 1)
				ans = (ans*a) % n;
			a = (a*a) % n;
			b = b / 2;
		}
		printf("%d\n",ans);
	}

	return 0;
}

附：快速幂取模详解！

http://www.cnblogs.com/PegasusWang/archive/2013/03/13/2958150.html

我们先从简单的例子入手：求a^b % c = ?

算法1.首先直接地来设计这个算法：

int ans = 1;

for(int i = 1;i<=b;i++)

{

ans = ans * a;

}

ans = ans % c;

这个算法的时间复杂度体现在for循环中，为O（b）.这个算法存在着明显的问题，如果a和b过大，很容易就会溢出。

那么，我们先来看看第一个改进方案：在讲这个方案之前，要先有这样一个公式：a^b%c=(a%c)^b%c.这个公式大家在离散数学或者数论当中应该学过，不过这里为了方便大家的阅读，还是给出证明：

引理1：a^b%c = (a%c)^b%c

 

上面公式为下面公式的引理，即积的取余等于取余的积的取余。

 

证明了以上的公式以后，我们可以先让a关于c取余，这样可以大大减少a的大小，

于是不用思考的进行了改进：

算法2：

int ans = 1;

a = a % c; //加上这一句

for(int i = 1;i<=b;i++)

{

ans = ans * a;

}

ans = ans % c;

聪明的读者应该可以想到，既然某个因子取余之后相乘再取余保持余数不变，那么新算得的ans也可以进行取余，所以得到比较良好的改进版本。

算法3：

int ans = 1;

a = a % c; //加上这一句

for(int i = 1;i<=b;i++)

{

ans = (ans * a) % c;//这里再取了一次余

 

}

ans = ans % c;

这个算法在时间复杂度上没有改进，仍为O(b)，不过已经好很多的，但是在c过大的条件下，还是很有可能超时，所以，我们推出以下的快速幂算法。

快速幂算法依赖于以下明显的公式，我就不证明了。

 

那么我们可以得到以下算法：

算法4：

int ans = 1;

a = a % c;

if(b%2==1)

ans = (ans * a) mod c; //如果是奇数，要多求一步，可以提前算到ans中

k = (a*a) % c; //我们取a2而不是a

for(int i = 1;i<=b/2;i++)

{

ans = (ans * k) % c;

}

ans = ans % c;

 

我们可以看到，我们把时间复杂度变成了O(b/2).当然，这样子治标不治本。但我们可以看到，当我们令k = (a * a) mod c时，状态已经发生了变化，我们所要求的最终结果即为(k)b/2 mod c而不是原来的ab mod c，所以我们发现这个过程是可以迭代下去的。当然，对于奇数的情形会多出一项a mod c，所以为了完成迭代，当b是奇数时，我们通过

ans = (ans * a) % c;来弥补多出来的这一项，此时剩余的部分就可以进行迭代了。

 

形如上式的迭代下去后，当b=0时，所有的因子都已经相乘，算法结束。于是便可以在O（log b）的时间内完成了。于是，有了最终的算法：快速幂算法。

算法5：快速幂算法

 

int ans = 1;

a = a % c;

while(b>0)

{

 

if(b % 2 == 1)

ans = (ans * a) % c;

b = b/2;

a = (a * a) % c;

}

将上述的代码结构化，也就是写成函数：

int PowerMod(int a, int b, int c)

{

int ans = 1;

a = a % c;

while(b>0)

{

 

if(b % 2 = = 1)

ans = (ans * a) % c;

b = b/2;

a = (a * a) % c;

}

return ans;

}

本算法的时间复杂度为O（logb），能在几乎所有的程序设计（竞赛）过程中通过，是目前最常用的算法之一。