a^b

题目描述

求 a 的 b 次方对 p 取模的值。

输入格式

三个整数 a、b、p ，在同一行用空格隔开。

输出格式

输出一个整数，表示 a^b mod p的值。

数据范围

$1\le a,b,p\le 10^9$

输入样例：

3 2 7

输出样例：

2

解题思路

本题目就是一道快速幂乘的题目。

这里再次解释一下快速幂乘的原理。

假定求解 $a^b$ 的值，并且 b ⇔ k n k n − 1 … … k 0 ( k i ∈ { 0 , 1 } ) b \Leftrightarrow k_nk_{n-1}……k_0 (k_i \in \{0,1\}) b⇔kn​kn−1​……k0​(ki​∈{0,1}) 。

可以得到如下推论：$$a^b\iff a^{k_n{k}_{n-1}\dots \dots k_0}\iff a^{2^n\times k_n+2^{n-1}\times k_{n-1}+\dots \dots +2^0\times k_0}\iff a^{2^n\times k_n}\times a^{2^{n-1}\times k_{n-1}}\times \dots \dots \times a^{2^0\times k_0}$$

对于任意 $a^{2^n\times k_n}$，如果 $k_n==0$ ，则 $a^{2^n\times k_n}=1$ 。如果 $k_n==1$ ，则 $a^{2^n\times k_n}=a^{2^{n-1}}\times a^{2^{n-1}}$ 。

我们可以依次记录所有的 $a^{2^i}$ ，然后根据对应的 $k_i$ 得到 $a^b$ 的值。

本题目唯一需要注意一点的就是数值溢出。

#include <iostream>

using namespace std;

int main()
{
    long a,b,p;
    cin >> a >> b >> p;
    long result = 1 % p,base = a;
    while(b)
    {
        if(b & 1)
        {
            result = (result * base) % p;
        }
        b = b >> 1;
        base = (base * base) % p;
    }
    
    cout << result << endl;
    
    return 0;
}