快速幂算法

快速幂

给定n组$a_i,k_i,p_i$,对于每组数据，求出$a_i^{k_i}$mod$p_i$。我们的快速幂算法是这样子的，首先我考虑将$k$写成二进制的形式，那么这个表示形式一定是可以的，而且是唯一的。那么这样一来假设$k$=$2^{b_1}+2^{b_2}+...+2^{b_n}$，那么$a^k$就可以写成$a^{2^{b_1}+2^{b_2}+...+2^{b_n}}$=$a^{2^{b_1}}\ast a^{2^{b_2}}\ast ...\ast a^{2^{b_n}}$，那么$a^{k}modp$就可以写成$a^{2^{b_1}}modp\ast a^{2^{b_2}}modp\ast ...\ast a^{2^{b_n}}modp$,（这是有定理可以证明的)。那么现在的关键是如何确定$b_1,b_2,...,b_n$,其实很简单，就是直接用$k\&1$即可然后，然后$k>>=1$,及$k$右移一位，那么怎么计算$a^{2^{b_i}}$呢，我们可以这样来看，$a^{2^0}=a^1,a^{2^1}=a^2=a\ast a,a^{2^2}=a^4=a^2\ast a^2,a^{2^3}=a^8=a^4\ast a^4...a^{2^i}=a^{2^{i-1}}\ast a^{2^{i-1}}$，根据这一个性质，我们只需要每次对$k$进行右移的同时对$a$进行$a=a\ast a$的计算即可，不过要注意$modp$处理，同时由于中间过程结果可能过大，所以要把中间过程的结果转换成$longlong$形式，接下来看一下代码:

#include<iostream>
#include<algorithm>

using namespace std;

typedef long long LL;

int qmi(int a, int k, int p)
{
    int res = 1;
    
    while(k)
    {
        if(k & 1) res = (LL)res * a % p;
        k >>= 1;
        a = (LL)a * a % p;
    }
    
    return res;
}

int main()
{
    int n;
    cin >> n;
    
    while(n --)
    {
        int a, k, p;
        
        scanf("%d%d%d", &a, &k, &p);
        printf("%d\n", qmi(a, k, p));
    }
    
    return 0;
}

第一次用LaTex写公式，虽然麻烦了些，但写出来真的好看。
$\mathbb{Y}\mathbb{Y}\mathbb{D}\mathbb{S}$
$\mathcal{Y}\mathcal{Y}\mathcal{D}\mathcal{S}$