洛谷刷题之P1226

【模板】快速幂

题目描述

给你三个整数 a , b , p a,b,p a,b,p，求 a b   m o d   p a^b \bmod p abmodp。

输入格式

输入只有一行三个整数，分别代表 a , b , p a,b,p a,b,p。

输出格式

输出一行一个字符串 a^b mod p=s，其中 a , b , p a,b,p a,b,p 分别为题目给定的值， s s s 为运算结果。

样例 #1

样例输入 #1

2 10 9

样例输出 #1

2^10 mod 9=7

提示

样例解释

2 10 = 1024 2^{10} = 1024 210=1024， 1024   m o d   9 = 7 1024 \bmod 9 = 7 1024mod9=7。

数据规模与约定

对于 100 % 100\% 100% 的数据，保证 0 ≤ a , b < 2 31 0\le a,b < 2^{31} 0≤a,b<231， a + b > 0 a+b>0 a+b>0， 2 ≤ p < 2 31 2 \leq p \lt 2^{31} 2≤p<231。

分析

这道题目就是一道板子题目

Code

#include <bits/stdc++.h>
#define int long long

using namespace std;

int pos(int a, int b, int p)
{
    int ans = 1;
    while (b > 0)
    {
        if (b % 2 == 1)
        {
            ans *= a;
            ans %= p;
        }
        a *= a;
        a %= p;
        b /= 2;
    }
    return ans;
}

signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    int a, b, p;
    cin >> a >> b >> p;
    cout << a << "^" << b << " mod " << p << "=" << pos(a, b, p) << endl;
    return 0;
}