矩阵乘法|蓝桥杯

Description

给定一个N阶矩阵A，输出A的M次幂（M是非负整数）

例如：

A =

1 2

3 4

A的2次幂

7 10

15 22

Input

输入描述:

第一行是一个正整数N、M（1<=N<=30, 0<=M<=5），表示矩阵A的阶数和要求的幂数

接下来N行，每行N个绝对值不超过10的非负整数，描述矩阵A的值

输入样例:

2 2

1 2

3 4

Output

输出描述:

输出共N行，每行N个整数，表示A的M次幂所对应的矩阵。相邻的数之间用一个空格隔开

输出样例:

7 10

15 22

思路

memset(tmp,0,sizeof tmp);用于初始化较大的矩阵

//矩形面积交
#include<iostream>
#include<string.h>

using namespace std;
int n,m;    

int main(){
    cin>>n>>m;
    int lst[n][n]; 
    
    for (int i=0;i<n;i++)
    for (int j=0;j<n;j++)
    cin>>lst[i][j];
    
    int tmp[n][n]; //保存最初的数组
    memset(tmp,0,sizeof tmp); 
    
    for (int i=0;i<n;i++){
        for (int j=0;j<n;j++){
            tmp[i][j]=lst[i][j];
        }
    }
    
    if (m==0){
        for (int i=0;i<n;i++){
            for (int j=0;j<n;j++){
                (i==j) ? cout<<1<<" " : cout<<0<<" "; 
            }
            cout<<endl; 
        }
    }
    else{
        for (int i=1;i<m;i++){
        
        int res[n][n]; //记录正在变化的数组，矩阵相乘是所有乘完才会变化
        memset(res,0,sizeof res); 
        
        for (int i=0;i<n;i++)
        for (int j=0;j<n;j++)
        for (int k=0;k<n;k++)
        res[i][j]+=lst[i][k]*tmp[k][j];
        
        for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
        lst[i][j]=res[i][j];
        
    }
    
    for (int i=0;i<n;i++){
        for (int j=0;j<n;j++){
            cout<<lst[i][j]<<" "; 
        }
        cout<<endl; 
    } 
    }
    return 0;
}