暴力算法
我们可以采用暴力算法
#include<bits/stdc++.h>
using namespace std;
#define ll long long
int main()
{
ll a, b, c;
cin >> a >> b >> c;
ll ans = 1;
for (ll i = 1; i <= b; i++) {
ans *= a;
}
ans %= c;
cout << ans;
}
不过这样肯定会超时的
可以改进吗
我们注意到 (a * b) % c = (a % c + b % c ) % c
鉴于这个定理
我们可以每次for循环中 对 ans 取模一次
ans *= a
ans %= c
不过时间复杂度还是没有优化
看看代码实现
#include<bits/stdc++.h>
using namespace std;
#define ll long long
int main()
{
ll a, b, c;
cin >> a >> b >> c;
ll ans = 1;
while (b) {
if (b % 2 == 1) {
ans = (ans * a) % c;
a = (a * a)%c;
b /= 2;
}
else {
a = (a * a)%c;
b /= 2;
}
}
cout << ans;
}
我们还可以简化代码
#include<bits/stdc++.h>
using namespace std;
#define ll long long
int main()
{
ll a, b, c;
cin >> a >> b >> c;
ll ans = 1;
a %= c;
while (b) {
if (b & 1) {
ans = (ans * a) % c;
}
a = (a * a) % c;
b >>= 1;
}
cout << ans;
}