python蓝桥杯经典题目：矩形的乘法

一、题目

问题描述

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

例如:

A=

12

34
|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代码与解析

 代码：

# coding=gbk

# 作者：终会为一
# @时间：2023/10/7

def solve(N, rect, rect_ans):  #定义矩阵相乘的函数

    rect2 = [[0 for _ in range(N)] for _ in range(N)]
    for i in range(N):
        for j in range(N):
            for n in range(N):
                rect2[i][j] += rect[i][n] * rect[n][j]
    return rect2


N, M = map(int, input().split())  # 输入数据
rect = [[] for _ in range(N)]
for i in range(N):
    arr = input().split()
    for j in range(N):
        rect[i].append(int(arr[j]))

if M > 0:
    rect_ans = rect
    for i in range(M - 1):  # 矩阵的M幂
        rect_ans = solve(N, rect, rect_ans)
else:  # 矩阵的0次幂为单位矩阵
    rect_ans = [[0 for _ in range(N)] for _ in range(N)]
    for i in range(N):
        rect_ans[i][i] = 1

for i in range(N):  # 格式化输出
    for j in range(N):
        print(rect_ans[i][j], end=" ")
    print()