python代码

def matmul(m1,m2):
    res=[[0]*len(m2[0]) for _ in range(len(m1))]
    for x in range(len(m1)):
        for y in range(len(m2[0])):
            n=0
            for a,b in zip(m1[x],(line[y] for line in m2)):
                n+=a*b
            res[x][y]=n
    return res


if __name__=='__main__':
    n,k=map(int,input().split())
    res=[[0]*n for _ in range(n)]
    for i in range(n):
        res[i][i]=1
    m=[]
    for i in range(n):
        m.append(list(map(int,input().split())))
    while k!=0:
        if k&1==1:
            res=matmul(m,res)
        m=matmul(m,m)
        k>>=1
    for i in range(n):
        for j in range(n):
            print(res[i][j],end=' ')
        print()