蓝桥杯 基础练习 矩阵乘法（Python）

资源限制

内存限制：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

import copy

N, M = map(int, input().split())
list1 = []
for i in range(N):
    list1.append(list(map(int, input().split())))
ans = copy.deepcopy(list1)  # 此处必须用深拷贝，否则改动一个，则另一个也会改变，一维数组的深拷贝用 .copy()或者[:]就行，二维数组则用copy.deepcopy()
if M == 0:  # 如果是一次幂，则将ans赋值成单位矩阵
    for i in range(N):
        for j in range(N):
            if i == j:
                ans[i][j] = 1
            else:
                ans[i][j] = 0
else:  # 如果不是0次幂，则一次一次乘到目标次幂
    li = copy.deepcopy(list1)
    for i in range(1, M):
        li = copy.deepcopy(ans)
        for j in range(N):
            for k in range(N):
                num = 0
                for m in range(N):
                    num += list1[j][m] * li[m][k]
                ans[j][k] = num
for i in range(N):  # 按规则输出结果即可
    for j in range(N):
        if j != N - 1:
            print(ans[i][j], end=" ")
        else:
            print(ans[i][j])