快速幂运算的应用

快速幂取余的方法，所得返回值为 a^b % p

int f(int a,int b,int p)
{
    int temp = 1;
    while(b)
    {
        if(b & 0x01)
        {
            temp = (temp * (a%p)) % p;
        }
        a = ( (a%p) * (a%p) ) % p;
        b >>= 1;
    }
    return temp;
}

例如：a^11，设b = 11，写成2进制为1011。

则a^11可以表示为   a^( 1*2^3 + 0*2^2 + 1*2^1 + 1*2^0 ) = a^(2^3) *a^(2^1) * a^(2^0)

二进制值为1的那项，在a的指数中就存在a^(2^i)，

所以可以把a^(2^0),a^(2^1),a^(2^2)......求出来，

如果在b的二进制表示中第i位为1，就把a^(2^i)乘到最终保存结果的变量中去。

而求a^(2^0),a^(2^1),a^(2^2)......a^(2^i)......时，

a^(2^ (i+1) ) = a^( 2^i * 2 ) = (a^(2i))^2

所以如果知道了a^(2^0) ，对它进行平方就可以得到a^(2^1)，以此类推，

所以在程序中我们就不断的将a平方，再将b向右移位。

Q1：Compute (a^(b^c)) % 317000011   ( 1 <= a,b,c <= 100000)

Tips:

⒈ 费马小定理: 如果p为素数，则a^(p-1) % p = 1;

⒉ 除法定理: 对任意整数a和任意正整数n，存在唯一的整数q和r，满足 0 <= r < n ， 并且 a = qn + r;

解：设b^c = n(p-1) + d，则a^(b^c) = a^(n(p-1)) * a^(d),依据定理⒈ ，a^(b^c) = a^(d),

而d = b^c Mod (p-1)。所以 a^(b^c) % p = a^( b^c Mod (p-1) ) Mod p 。

#include <stdio.h>

#define P 317000011

unsigned long long f(unsigned long long a,unsigned long long b,unsigned long long p)
{
    unsigned long long temp = 1uLL;
    while(b)
    {
        if(b & 0x01uLL)
        {
            temp = ((temp%p) * (a%p)) % p;
        }
        a = ((a%p)*(a%p))%p;
        b >>= 1uLL;
    }
    return temp;
}


int main()
{
    int t;
    unsigned long long a,b,c;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%llu %llu %llu",&a,&b,&c);
        printf("%llu\n",f(a,f(b,c,P-1),P));
    }
    return 0;
}

Q2：矩阵幂 快速求Fibonacci数列的第 N 项

定义一个基础矩阵 base

┌      ┐     ┌      ┐

│ 1  1 │   = │F2  F1│

│ 1  0 │     │F1  F0│

└      ┘     └      ┘

             ┌          ┐

求base^n  =  │Fn+1  Fn  │

             │Fn    Fn-1│

             └          ┘

所以矩阵的快速幂乘方方法，可以高效的求的Fibonacci数列，O(log2n)。

int f(int n)
{
    if(n == 0)  return 0;
    else if(n == 1) return 1;
    else if(n == 2) return 1;
    else
    {
        int base[2][2] = {{1,1},{1,0}};
        int b = n-1;
        int res[2][2] = {{1,1},{1,1}};
        int temp[2][2];
        while(b)
        {
            if(b & 0x01)
            {
                temp[0][0] = res[0][0] * base[0][0] + res[0][1] * base[1][0];
                temp[0][1] = res[0][0] * base[0][1] + res[0][1] * base[1][1];
                temp[1][0] = res[1][0] * base[0][0] + res[1][1] * base[1][0];
                temp[1][1] = res[1][0] * base[0][1] + res[1][1] * base[1][1];
                res[0][0] = temp[0][0]; res[0][1] = temp[0][1];
                res[1][0] = temp[1][0]; res[1][1] = temp[1][1];
            }
            temp[0][0] = base[0][0] * base[0][0] + base[0][1] * base[1][0];
            temp[0][1] = base[0][0] * base[0][1] + base[0][1] * base[1][1];
            temp[1][0] = base[1][0] * base[0][0] + base[1][1] * base[1][0];
            temp[1][1] = base[1][0] * base[0][1] + base[1][1] * base[1][1];
            base[0][0] = temp[0][0]; base[0][1] = temp[0][1];
            base[1][0] = temp[1][0]; base[1][1] = temp[1][1];
            b >>= 1;
        }
        return res[0][0];
    }
}