基础训练 矩阵乘法

Posted 2021-02-07 a-little-nut

矩阵乘法

#include<iostream>
#include<vector>
using namespace std;
int n, m;
vector<vector<int> > Mul(vector<vector<int> > t1, vector<vector<int> > t2){
    vector<vector<int> > temp(n, vector<int>(n, 0));
    for(int i=0; i<n; i++)
        for(int j=0; j<n; j++)
            for(int k=0; k<n; k++)
                temp[i][j]+=t1[i][k]*t2[k][j];
    return temp;    
} 
vector<vector<int> > QuickMul(vector<vector<int> > A, vector<vector<int> > ans, int m){
    while(m){
        if(m&1)
            ans=Mul(ans, A);
        A=Mul(A, A);
        m=m>>1;
    }
    return ans;
}
int main(){
    cin>>n>>m;
    vector<vector<int> > ans(n, vector<int>(n,0)), A(n, vector<int>(n,0));
    for(int i=0; i<n; i++)
        for(int j=0; j<n; j++)
            cin>>A[i][j];
    for(int i=0; i<n; i++)
        for(int j=0; j<n; j++)
            if(i==j)
                ans[i][j]=1;
    ans=QuickMul(A, ans, m); 
    for(int i=0; i<n; i++){
        int flag=0;
        for(int j=0; j<n; j++)
            if(flag++==0) cout<<ans[i][j];
            else cout<<" "<<ans[i][j];
        cout<<endl; 
    }
    return 0; 
}