试题 基础练习 矩阵乘法

资源限制

内存限制：512.0MB   C/C++时间限制：1.0s   Java时间限制：3.0s   Python时间限制：5.0s

问题描述

　　给定一个N阶矩阵A，输出A的M次幂（M是非负整数）
　　例如：
　　A =
　　1 2
　　3 4
　　A的2次幂
　　7 10
　　15 22

输入格式

　　第一行是一个正整数N、M（1<=N<=30, 0<=M<=5），表示矩阵A的阶数和要求的幂数
　　接下来N行，每行N个绝对值不超过10的非负整数，描述矩阵A的值

输出格式

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

样例输入

2 2
1 2
3 4

样例输出

7 10
15 22

#include<iostream>
using namespace std;

const int N = 1e5 + 10;

int re[31][31];//结果矩阵

int main(){
    int m,n;
    cin>>m>>n;
    int a[m][m],z[m][m];//原矩阵，中间备份矩阵
    for(int i=0;i<m;i++){
        for(int j=0;j<m;j++){
            cin>>a[i][j];
            z[i][j]=a[i][j];
        }
    }
    if(m==1&&a[0][0]==1){//如果是1*1 的矩阵且元素为1，则无论多少次幂都是1
        cout<<1<<endl;
        return 0;
    }
    else if(n==0){//如果是求0次幂，任意矩阵的0次幂都是单位矩阵，输出单位矩阵即可
        for(int i=0;i<m;i++)//初始化主对角线元素为1
            re[i][i]=1;
        for(int i=0;i<m;i++){
            for(int j=0;j<m;j++)
                cout<<re[i][j]<<" ";
            cout<<endl;
        }
        return 0;
    }
    n--;//求n次幂只需计算n-1次乘法即可
    while(n--){
        for(int i=0;i<m;i++){
            for(int j=0;j<m;j++){
                re[i][j]=0;
                for(int k=0;k<m;k++){
                    re[i][j]+=z[i][k]*a[k][j];
                }
            }
        }
        for(int i=0;i<m;i++){//把每一次乘法后的结果矩阵赋值给中间备份矩阵，用于下一次与原矩阵相乘
            for(int j=0;j<m;j++){
                z[i][j]=re[i][j];
            }
        }
    }
     for(int i=0;i<m;i++){
        for(int j=0;j<m;j++){
            cout<<re[i][j]<<" ";
        }
        cout<<endl;
    }
    return 0;
}