蓝桥杯 BASIC-17 矩阵乘法 Python实现

问题描述

给定一个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

代码

python.

N,M = input().split()
N = int(N)
M = int(M)


def chengfa(a,b,N):
    res2=[]
    for i in range(N):#a矩阵第i行
        res1 = []
        for j in range(N):# b矩阵第j列
            sum_1 = 0
            for k in range(N):# 第k元素对应相乘
                sum_1 += a[i][k]*b[k][j]
            res1.append(sum_1)
        res2.append(res1)
    return res2

def E(N):
    res2 = []
    for i in range(N):
        res1 = [0 for x in range(N)]
        res1[i]=1
        res2.append(res1)
    return res2

arr= []
for i in range(N):
    arr.append(list(map(int,input().split())))


res = arr
if M==0:
    res = E(N)
else:
    for i in range(M-1):
        res = chengfa(res,arr,N)
        
for i in range(N):
    print(*res[i])