洛谷刷题之P1226

【模板】快速幂

题目描述

给你三个整数 $a,b,p$，求 $a^b\mathrm{mod}p$。

输入格式

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

输出格式

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

样例 #1

样例输入 #1

2 10 9

样例输出 #1

2^10 mod 9=7

提示

样例解释

$2^{10}=1024$，$1024\mathrm{mod}9=7$。

数据规模与约定

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

分析

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

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;
}