본문 바로가기

프로그래머스

프로그래머스 LV2 행렬의 곱셈

728x90
def solution(arr1, arr2):
    answer = []
    for i in range(len(arr1)): 
        row = []
        for j in range(len(arr2[0])): 
            value = 0
            for k in range(len(arr2)): 
                value += arr1[i][k] * arr2[k][j]
            row.append(value)
        answer.append(row)
    return answer

 

풀이 및 회고

우선, 행렬의 곱을 하기 위해선 행렬의 곱 공식을 알아야할 수 있다. 첫번째 열과 두번째 행 요소들 끼리 각각 곱한 것을 더해주고 행에 추가해주는 원리(?) 입니다.

 

728x90