《算法竞赛进阶指南》0x01基本算法 快速幂

来源：AcWing 89:a^b
题目描述：求a的b次方对p取模的值，其中1≤a,b,p≤10^9
困惑点：直接采用for循环来算a^b，时间复杂度为O(b)，当b比较大时，就容易超时
解决办法：采用快速幂算法，将时间复杂度转换为O($lo{g}_{2}n$)
关键字：快速幂、位运算

快速幂算法思路：

实现代码：

#include <iostream>
using namespace std;

int power(int a, int b, int p){    //calculate （a^b）mod p
    int ans = 1 % p;    //ans作为结果
    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,res;
    cin>>a>>b>>p;
    res = power(a,b,p);
    cout<<res;
    return 0;
}

yxc快速幂算法模板