矩阵快速幂模板

贴一个矩阵快速幂的模板,具体的讲解可以去看这里啦一个关于一些快速幂的视频讲解

重新整理了一下模板,发上来了

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

using namespace std;
typedef long long ll;
const ll maxn = ;//矩阵单位
const ll MOD = 1e9+7;
#define mod(x) ((x)%MOD)

struct mat{
    ll m[maxn][maxn];
    /*
    void Clear(){memset(m,0,sizeof(m));}
    void unit(){
        Clear();
        for(int i = 0;i <= maxn-1;i++) m[i][i] = 1;
    }//有时候可能会用到去简化处理
    */
}unit;

mat operator *(mat a,mat b){
    mat ret;
    ll x;
    for(ll i = 0;i < maxn;i++){
        for(ll j = 0;j < maxn;j++){
            x = 0;
            for(ll k = 0;k < maxn;k++)
                x += mod((ll)a.m[i][k]*b.m[k][j]);
            ret.m[i][j] = mod(x);
        }
    }
    return ret;
}//普通处理

/*
mat operator *(mat a,mat b){
    mat ret;
    for(ll i = 0;i < maxn;i++)
        for(ll j = 0;j < maxn;j++)
            ret.m[i][j] = 0;
    for(ll i = 0;i < maxn;i++){
        for(ll k = 0;k < maxn;k++){
            if(a.m[i][k]){
                for(ll j = 0;j < maxn;j++)
                    ret.m[i][j] = mod(ret.m[i][j] + mod(a.m[i][k]*b.m[k][j]));
            }
        }
    }
    return ret;
}稀疏矩阵剪枝处理
*/

void init_unit(){
    for(ll i = 0;i < maxn;i++)
        unit.m[i][i] = 1;
    return ;
}

mat pow_mat(mat a,ll n){
    mat ret = unit;
    while(n){
        if(n&1) ret = ret*a;
        a = a*a;
        n >>= 1;
    }
    return ret;
}

int main(){
    ll N,M,x;
    init_unit();
    while(~scanf("%lld%lld",&N,&M)){

        mat a,b;
        //对a,b矩阵进行初始化的处理
        b = pow_mat(b,M);//多少次幂由题意决定
        a = a*b;
        printf("%lld\n",(a.m[0][0]+MOD)%MOD);//输出由题意决定


    }
    return 0;
}



你可能感兴趣的:(模板)