题目链接:https://www.luogu.org/problem/P1226

洛谷:快速幂模板题

题目描述

输入b,p,k的值,求b^p mod k的值。其中b,p,k*k为长整型数。

输入格式

三个整数b,p,k.

输出格式

输出“b^p mod k=s”
s为运算结果

思路

自定义快速幂函数可用二分思想对b进行处理,如果b为奇数,则b=b/2 * b/2 *b,b为偶数,则b=b/2 * b/2。

上代码

#include <bits/stdc++.h>
using namespace std;
#define LL long long
LL qmi(int a,int b,int p)
{
    LL ans=1;
    while(b)
    {
        if(b&1)ans=ans*a%p;
        a=a*(LL)a%p;
        b>>=1;
    }
    return ans;
}
int main()
{
    int a,b,p;
    cin>>a>>b>>p;
    printf("%d^%d mod %d=",a,b,p);
    if(b==0)cout<<0<<endl;
    else cout<<qmi(a,b,p)<<endl;
    return 0;
}