백준

행렬의 곱 구현코드 리스트버전

백준파이썬개발자:프로젝트골드 2024. 1. 8. 04:08
반응형
def mul(U, V):
    n = len(U)
    Z = [[0] * n for _ in range(n)]

    for row in range(n):
        for col in range(n):
            e = 0
            for i in range(n):
                e += U[row][i] * V[i][col]
            Z[row][col] = e % 1000

    return Z
반응형