矩阵快速幂-模板

矩阵快速幂-模板

&思考：

这个就是可以用来解开m像递推式, 白书P200

&代码：

typedef vector<int> vec;
typedef vector<vec> mat;

mat mul(mat &A,mat& B)
{
    //假设矩阵A是m*n B是n*h 那么C=A*B 是m*h
    mat C(A.size(),vec(B[0].size()));
    for(int i=0;i<A.size();i++)
    for(int k=0;k<B.size();k++)
    for(int j=0;j<B[0].size();j++){
        C[i][j]=(C[i][j]+A[i][k]*B[k][j]);
    }
    return C;
}

//求矩阵A的n次方 即A^n
mat bin_pow(mat A,ll n)
{
    //单位矩阵B
    mat B(A.size(),vec(A.size());
    for(int i=0;i<A.size();i++){
        B[i][i]=1;
    }
    while(n>0){
        if(n&1) B=mul(B,A);
        A=mul(A,A);
        n>>=1;
    }
    return B;
}

转载于:https://www.cnblogs.com/s1124yy/p/6579583.html