蓝桥杯-基础练习VIP-矩阵乘法-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 = map(int, input().split())
arr = [list(map(int, input().split()))for _ in range(n)]
xx=[[] for i in range(n)]
def multi(res, arr, m):
    c=[[] for i in range(n)]
    if m == 1:
        return res
    else:
        for i in range(n):
            lst1 = []
            lst1=[res[i][x] for x in range(n)]
            for j in range(n):
                lst2 = []
                lst2=[arr[y][j] for y in range(n)]
                sum = 0
                for z in range(n):
                    sum += (lst1[z]*lst2[z])
                c[i].append(sum)           
        res = multi(c, arr, m-1)
        return res
if m ==0:
    for i in range(n):
        for j in range(n):
            if i == j :
                print('1', end=' ')
            else:
                print('0', end=' ')
        print('')
else:
    res = arr
    xx=multi(res, arr, m)
    for i in range(n):
        for j in range(n):
            print(xx[i][j], end=' ')
        print('')