快速幂的学习

前言

顾名思义，快速幂就是快速算底数的n次幂。其时间复杂度为 O(log₂N)， 与朴素的O(N)相比效率有了极大的提高。
上述是百度百科的解释，快速幂就是在logn的复杂度情况下计算底数n次幂的一种算法。快速幂的板子很短，也很容易记住，在此分享自己对于快速幂的一种理解。

朴素的O(n)算法

代码：

int pow(int A,int n){
    int rst = 1;
    for (int i = 1; i <= n; i++) {
        rst *= A;
    }
    return rst;
}

相信初学者都是如此求幂的，其复杂度O(n)，对于数据量大时便不再适用。

二分思想O(log2n)的快速幂算法

我们学习复杂度时，看到logn级别的复杂度，第一时间想到的便是二分思想，快速幂也是一种二分思想减小复杂度的算法。

代码 ：

int fastpow(int a,int k){
    if (k == 0) {
        return a;
    }
    int ans = 1;
    while (k) {
        if(k & 1) ans *= a;
        a *= a;
        k>>=1;
    }
    return ans;
}

举个例子来具体说明此代码： 3^11
首先，将11写作二进制，即1011.则

所以代码中的 (k&1) 即是取最后一位，如果是1，就 *a . 而每次 a *= a，则是 每位对应的 3 ^ 1 , 3 ^ 2 , 3 ^ 4……
而代码中的 k>>=1 ，则是右移一位，保证每次取的都是新的最后一位。
我们打表来观察一下计算 3^11的各个变量的具体值：

可以比较清晰的看出其中的每一步的变化。

矩阵快速幂

首先是简单的矩阵A的k次方的计算。从上面的整数的k次方计算可知，矩阵的k次方也是按照二进制计算，不同的是需要重载矩阵的乘法运算即可。
1、矩阵结构体的定义

struct Matrix{
    int N;
    int **m;

    void clear(){					
        for (int i = 0; i < N; i++) {
            memset(m[i],0,sizeof(int)*N);
        }
    }
    Matrix(int n){  		//相当于构造方法，构造一个n*n的矩阵
        m = new int*[n];
        for (int i = 0; i < n; i++) {
            m[i] = new int[n];
        }
        N = n;
        clear();
    }
    void unit(){			//把矩阵化为单位矩阵
        clear();
        for (int i = 0; i < N; i++) {
            m[i][i] = 1;
        }
    }
    void display(){			//打印出矩阵来
        for (int i = 0; i < N; i++) {
            cout<<m[i][0];
            for (int j = 1; j < N; j++) {
                cout<<" "<<m[i][j];
            }
            cout<<endl;
        }
    }
    void operator = (const Matrix& tmp) const {		//重载 = 运算符
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                m[i][j] = tmp.m[i][j];
            }
        }
    }
    Matrix operator * (const Matrix& tmp)const{		//重载 * 运算符，矩阵乘法 O(n3)
        Matrix ans(tmp.N);
        for (int i = 0; i < tmp.N; i++) {
            for (int j = 0; j < tmp.N; j++) {
                for (int k = 0; k < tmp.N; k++) {
                    ans.m[i][j] += m[i][k] * tmp.m[k][j];
                }
            }
        }
        return ans;
    }
};

2、接下来定义好矩阵结构体，并重载运算符之后，只要把快速幂里的 int a, 换为 Matrix a 即可。

Matrix fastpow(Matrix base,int k){
    Matrix ans(base.N);			// N * N的矩阵
    ans.unit();					// ans 化为单位矩阵，就相当于整数快速幂里的  ans = 1
    if (k == 0) {
        return base;
    }
    while (k) {
        if(k&1) ans = base*ans;
        base = base*base;
        k>>=1;
    }
    return ans;
}

矩阵快速幂的例题

HDU 5015
分析都大同小异，在此只放上代码以及出现的错误
1、代码：

#include<bits/stdc++.h>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
#define F first
#define S second
#define pb push_back
#define mk make_pair
typedef long long ll;
const int maxn = 10 + 5;
const int mod = 1e7 + 7;
struct Matrix{
    int row,col;
    ll mat[maxn][maxn];
    void display(){
        for (int i = 0; i < row; i++) {
            cout<<mat[i][0];
            for (int j = 1; j < col; j++) {
                cout<<" "<<mat[i][j];
            }
            cout<<endl;
        }
    }
};
Matrix mulMatric(Matrix a, Matrix b, int p){
    Matrix ans;
    ans.row = a.row;
    ans.col = b.col;
    mem(ans.mat,0);
    for (int i = 0; i < a.row; i++) {
        for (int j = 0; j < b.col; j++) {
            for (int k = 0; k < a.col; k++) {
                ans.mat[i][j] += a.mat[i][k] * b.mat[k][j];
                ans.mat[i][j] %= p;
            }
        }
    }
    return ans;
}
Matrix fastpow(Matrix a,int k,int p){
    if (k == 0) {
        return a;
    }
    Matrix ans;
    ans.row = a.row;
    ans.col = a.col;
    mem(ans.mat,0);
    for (int i = 0; i < a.row; i++) {
        ans.mat[i][i] = 1;
    }
    while (k) {
        if(k&1) ans = mulMatric(ans,a,p);
        a = mulMatric(a,a,p);
        k>>=1;
    }
    return ans;
}
int n,m;
int main(){
    // freopen("1.txt","r",stdin);
    while (~scanf("%d%d",&n,&m) && n) {
        Matrix A,B,ans;         //A是关系矩阵
        B.row = n+2; B.col = 1;
        B.mat[0][0] = 23; B.mat[n+1][0] = 3;
        for (int i = 1; i <= n; i++) {
            scanf("%lld",&B.mat[i][0]);
        }
        A.row = A.col = n+2;
        mem(A.mat,0);
        for(int i = 0;i < n+1;i++) A.mat[i][0] = 10;
        for(int i = 0;i <= n+1;i++) A.mat[i][n+1] = 1;
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= i; j++) {
                A.mat[i][j] = 1;
            }
        }
        ans = fastpow(A,m,mod);
        ans = mulMatric(ans,B,mod);
        cout<<ans.mat[n][0]<<endl;
    }
    return 0;
}

2、出现的错误
（1）特别特别要注意下标，因为此题中的关系矩阵是一个 (n+2) * (n+2) 的矩阵
（2）注意在快速幂函数，以及矩阵乘法函数中，每次声明 Matrix ans，都要给其行和列赋值。并且每次都要初始化，不然都不默认是0.
矩阵快速幂题目的难点一般是找关系方程，不会直接让你求一个矩阵的几次方。此题求幂和上面不同的是，乘法是矩阵乘法的一般形式，即 (n,k) 矩阵乘以 (k,m) 矩阵。除此之外还要注意的是取模运算的位置。