본문 바로가기

백준

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

반응형
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
반응형

'백준' 카테고리의 다른 글

1065백준  (0) 2024.02.16
백준 1929번  (1) 2024.02.01
백준 10815번  (1) 2024.01.31
행렬의 곱셈 백준2740  (1) 2024.01.08
input관련 정리  (1) 2024.01.08