[Programmers] Lv2 - 행렬의 곱셈

문제

https://school.programmers.co.kr/learn/courses/30/lessons/12949

프로그래머스

 

풀이

사실 '행렬의 곱셈' 원리만 알고있다면 쉬운 문제다.

하지만 선형대수를 수강했음에도 전혀 기억하지 못하는 개구리새끼...가 있다.

식은 굉장히 익숙한데 말이다ㅋㅋ

 

행렬의 곱셈 개념

우선 행렬 a1, a2 가 있다고 가정해보자.

a1의 행렬이 m x n, a2의 행렬이 n x p일때에만 행렬의 곱셈이 가능해진다.

다시말해 a1 x a2 를 계산하고 싶다면 a1의 열 == a2의 행 이어야하고,

a2 x a1 을 계산하고 싶다면 a2의 열 == a1의 행 이어야한다.

 

 

따라서 a1 x a2의 결과 행렬을 a3라 하였을때, a3는 m x p 크기의 행렬이 되는것이다.

 

그래서 곱셈을 어떻게하느냐.

a3의 3행2열값을 알고싶다,

→ a1의 3행값들 * a2의 2열값들

을 계산해주면 된다.

 

만약 a1의 3행이 [3, 2, 1], a2의 2열이 각각 [5], [1], [2] 가 된다면 3*5 + 2*1 + 1*2 의 값이 a3의 3행2열값이 되는것이다.

# 행렬 곱셈하는법!
# ret(11) == arr1(1n) * arr2(n1)
def solution(arr1, arr2):
    # ret = [[0] * len(arr2[0])] * len(arr1) => 초기화 문제.
    ret = [[0 for _ in range(len(arr2[0]))] for _ in range(len(arr1))]
    def matrix(row, col):
        mat_sum = 0
        for i in range(len(arr2)): # 처음에 len(arr1)로 했다가 런타임 오류. 당연함.
            # len(행렬1의 열) == len(행렬2의 행) 이어야 하므로 arr1[0]이나 arr2를 넣어야한다.
            mat_sum += arr1[row][i] * arr2[i][col]
        return mat_sum

    for row in range(len(arr1)):
        for col in range(len(arr2[0])):
            ret[row][col] = matrix(row, col)
    return ret

처음에 ret = [[0] * len(arr2[0])] * len(arr1) 로 적었더니 답이 [[29, 20, 14], [29, 20, 14], [29, 20, 14]] 로 나왔다.

뭐라더라... 아무튼 참조때문이었는데. 리스트 컴프리헨션으로 초기화 해줬더니 됐다.

 

근데 테스트 1번에서 딱! 걸려버렸다.

나의 멍청함이 순간 튀어나온것이었다^^;

아까도 말했다시피 행렬1의 열 == 행렬2의 행 이어야 한다.

따라서 중복되는 값인 행1열 이나 행2행 으로 반복문을 돌려야했었다.