a^b的问题

题目描述

求 a 的 b 次方对 p 取模的值，其中 0≤a,b≤10^9 , 0<p≤10^9

输入

三个用空格隔开的整数a,b和p。

输出

一个整数，表示a^b mod p的值。

样例输入 Copy

2 3 9

样例输出 Copy

8

利用循环求a的b次方，运行的时间较长；

下为优化后的解：

#include<iostream>
using namespace std;
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;
}
int main()
{
	int a, b, p;
	cin >> a >> b >> p;
	cout << power(a, b, p);
}