[알고리즘] 프로그래머스 - 행렬의 곱셈

June·2021년 3월 31일
0

알고리즘

목록 보기
149/260
post-custom-banner

[프로그래머스 - 행렬의 곱셈](프로그래머스 - 행렬의 곱셈)

내 풀이

def solution(arr1, arr2):
    answer = [[0 for _ in range(len(arr2[0]))] for _ in range(len(arr1))]
    def calculate(arr1_row, arr2_col):
        tmp = 0
        for k in range(len(arr1[0])):
            tmp += arr1[arr1_row][k] * arr2[k][arr2_col]

        return tmp

    for i in range(len(arr1)):
        for j in range(len(arr2[0])):
            answer[i][j] = calculate(i, j)
    return answer

행렬 곱셈의 원리를 그대로 구현하면 되는 문제이다.

0개의 댓글