问题 A: 【快速幂】a^b
时间限制: 1 Sec 内存限制: 128 MB
题目描述
求 a 的 b 次方对 p 取模的值,其中 1≤a,b,p≤109
输入
三个用空格隔开的整数a,b和p。
输出
一个整数,表示ab mod p的值。
样例输入 Copy
2 3 9
样例输出 Copy
8
正常来说,计算机每秒可运算10^7 ~ 10^8(以c++语言来说)
所以这题如果用循环一个个的来乘,最大有10^9的运算量,有可能超时
应采用快速幂算法,运算量则会降为log级别的!
b == 3 (1 * 2^1+ 1*2^0 = 3)
二进制为11,b & 1取出最低位,若是1则相应乘以a的几次方
代码:
#include<bits/stdc++.h>
using namespace std;
ll power(ll a,ll b,ll p)
{
int ans = 1,base = a;
while(b)//b不为0时执行
{
if(b & 1) ans = ans*base % p;//判断该位是否为1
base = base*base % p;
b >>= 1;//右移,考虑下一位
}
return ans;
}
int main()
{
ll a,b,p; cin >> a >> b >> p;
cout << power(a,b,p) << endl;
return 0;
}
问题 B: 64位整数乘法
时间限制: 1 Sec 内存限制: 128 MB
题目描述
求 a 乘 b 对 p 取模的值,其中 1≤a,b,p≤10^18。
输入
第一行a,第二行b,第三行p。
输出
一个整数,表示a*b mod p的值。
样例输入 Copy
2
3
9
样例输出 Copy
6
a = 2
b = 3
二进制:11
a * b = 2 * (1 * 2^1+ 1 * 2^0 )= 2 * 1 * 2^0 + 2 * 1 * 2^1 = 6
代码:
#include<bits/stdc++.h>
using namespace std;
int main()
{
ll a,b,p; cin >> a >> b >> p;
ll ans = 0;
if(a < b) swap(a,b);
while(b)
{
if(b & 1) ans = (ans+a) % p;
a = a*2 % p;//左一位则*2
b >>= 1;
}
cout << ans << endl;
return 0;
}