【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}$。

代码如下：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<istream>
#include<iomanip>
#include<ostream>
#include<list>
#include<vector>
#include<set>
#include<map>
#include<fstream>
#include<stack>
#include<ctime>
#include<deque>
#include<queue>
#include <sstream>
#include <numeric>
#pragma warning (disable:4996)

using namespace std;

long long power(long long a, long long n, long long p) {
    if (n == 0) {
        return 1;
    }
    long long res = power(a, n / 2, p);
    res = (res * res) % p;
    if (n % 2 == 1) {
        res = (res * a) % p;
    }
    return res;
}

int main() {
    long long a, n, p;
    cin >> a >> n >> p;
    long long res = power(a, n, p);
    cout << a<<"^"<< n<<" mod " <<p<<"=" << res << endl;
    return 0;
}