矩陣快速冪

計算過程中避免溢出,最好用long long來定義

有必要時用unsigned long long

水水的代碼如下,權當記錄

#include <iostream>
#include <cstdio>
#include <string.h>
#include <queue>
#include <vector>
#include <cstdlib>


using namespace std;
typedef vector<vector<int> >mat;
typedef vector<int> vec;
const int mod=1e9+7;
mat mul(mat a,mat b){
    mat c(a.size(),vec(b[0].size()));
    for(int i=0;i<a.size();i++)
    for(int j=0;j<b[0].size();j++)
    for(int k=0;k<b.size();k++)
    c[i][j]=((long long)c[i][j]+((long long)a[i][k]*b[k][j])%mod)%mod;
    return c;
}
mat q_m_pow(mat a,int m){
    if(m==1)return a;
    mat c=q_m_pow(a,m/2);
    if(m%2)
    return mul(a,mul(c,c));
    else
    return mul(c,c);
}
int main()
{
    int n,m;
    cin>>n>>m;
    mat a(n,vector<int>(n));
    int c;
    for(int i=0;i<n;i++)
    for(int j=0;j<n;j++)
    {
        scanf("%d",&c);
        a[i][j]=c;
    }
    a=q_m_pow(a,m);
    for(int i=0;i<n;i++){
    for(int j=0;j<n;j++)
    cout<<a[i][j]<<' ';
    cout<<endl;
    }
    return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章