蓝桥杯练习系统习题（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次幂所对应的矩阵。相邻的数之间用一个空格隔开

---

100分

n,m=list(map(int,input().split()))              #输入是几阶，几次幂
A = [[0 for i in range(n)] for j in range(n)]   #用户未输入时的空列表A
for i in range(n):                              #保证输入的列数
    List = input().split()                      #循环输入数字，以空格分隔
    for j in range(n):
        A[i][j] = int(List[j])                  #用输入的数字代替原来A中的0
#函数AA，矩阵的两次幂的计算
def AA(n, a, b):                                #n是参数n阶矩阵，ab是参数
    Jieguo = [[0 for i in range(n)] for j in range(n)]   #初始化的输出结果列表
    for i in range(n):
        for j in range(n):
        #以上两行是定位Jieguo这个初始列表中的数的位置
            for k in range(n):
                Jieguo[i][j]=a[i][k]*b[k][j]+Jieguo[i][j]
            #以上两行将初始Jieguo中的0换成计算后的数
    return  Jieguo


if m==0:
    Jieguo = [[0 for i in range(n)] for j in range(n)]
    for i in range(n):
        Jieguo[i][i]=1
else:
    Jieguo = A                   
    for i in range(m-1):         #循环计算阶乘的幂，m-1是因为上面函数AA为二次幂的计算
        Jieguo=AA(n, Jieguo, A)

def printA(Jieguo):                 #Jieguo是列表，此函数将列表转变成题目要求形式输出
    for i in range(n):
        for j in range(n):
            print(Jieguo[i][j],end=" ")         #end=" "不换行
        print("")       #输出空格，换到下一行
#引用函数
printA(Jieguo)