蓝桥杯 算法提高——快速幂

发现官网给的免费蓝桥杯VIP5月初就过期了，从今天开始每天抽空做些题目

试题

资源限制、时间限制：1.0s   内存限制：256.0MB

问题描述

　　给定A, B, P，求(A^B) mod P。

输入格式

　　输入共一行。
　　第一行有三个数，N, M, P。

输出格式

　　输出共一行，表示所求。

样例输入

2 5 3

样例输出

2

数据规模和约定

　　共10组数据
　　对100%的数据，A, B为long long范围内的非负整数，P为int内的非负整数。

解题报告：

太久没打算法题，连快速幂的幂都给模上了，错了2遍才发现。。。

不加快速乘也能过，但是为了掌握一下快速乘（刚学，原来这么简单），也写上去了

#include<bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
typedef long long ll;
ll kadd(ll a,ll b,ll p) {    ///快速乘 
    ll ans=0;
    while(b) {
        if(b&1) ans=(ans+a)%p;
        a=(a<<1)%p;
        b>>=1;
    }
    return ans;
}
ll kpow(ll a, ll b, ll p) {    ///快速幂 
    ll ans=1;
    while(b) {
        if(b&1) ans=kadd(ans,a,p);
        a=kadd(a,a,p);
        b>>=1;
    }
    return ans;
}
int main()
{
    ll a,b,p;
    cin>>a>>b>>p;
    ll res=kpow(a%p,b,p);
    cout<<res<<endl;
    return 0;
}