89. a^b【快速幂模板】

a^b

Description

求 aa 的 bb 次方对 pp 取模的值。

输入格式

三个整数 a,b,pa,b,p ,在同一行用空格隔开。

输出格式

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

数据范围

1≤a,b,p≤10⁹

输入样例：

 3 2 7

输出样例：

 2

题解

快速幂模板，交了结果WA了

需要注意的是 123456789 0 1 这个数据，1%1 = 0

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll quickpow(ll a,ll b,ll p){
    if(a == 0) return 0 ;
    ll ans = 1 % p ;//123456789 0 1
    while(b){
        if(b & 1) ans =ans * a % p;
        b >>= 1;
        a = a * a % p;
    }
    return ans;
}
int main(){
    ll a,b,p;
    cin >> a >> b >> p;
    cout << quickpow(a,b,p) << endl;
}