矩阵快速幂模板

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

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

#include <map>
#include <set>
#include <queue>
#include <stack>
#include <cmath>
#include <vector>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>

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;
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值