프로그래머스 (Lv 2) - 행렬의 곱셈

1) 문제 설명

• 2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱한 결과를 반환하는 함수, solution을 완성해주세요.

 

2) 제한 사항

• 행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.

 

• 행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.

 

• 곱할 수 있는 배열만 주어집니다. (중요!)

 

3) 입출력 예제

arr1    arr2    return
[[1, 4], [3, 2], [4, 1]]    [[3, 3], [3, 3]]    [[15, 15], [15, 15], [15, 15]]
[[2, 3, 2], [4, 2, 4], [3, 1, 4]]    [[5, 4, 3], [2, 4, 1], [3, 1, 1]]    [[22, 22, 11], [36, 28, 18], [29, 20, 14]]

 

4) 입출력 예제 설명

• 없음

 

5) 풀이

def solution(arr1, arr2):
    # answer List 선언
    answer = []

    # 행렬 곱셈의 원리
    # 첫 번째 배열의 col 수 == 두 번쨰 배열의 row 수

    # 입출력 예시 1번
    # 3 x "2" 배열 * "2" x 2 배열 => "2" x "2" 배열

    # arr1 - 첫 번째 배열의 row 수만큼 반복
    for arr in arr1:
        L = []

        # arr2 - 두 번째 배열의 col 수만큼 반복
        for dx in range(len(arr2[0])):
            sum = 0

            # sum 함수에 곱셈 연산 실행한 것을 add
            for dy, num in enumerate(arr):
                sum += num * arr2[dy][dx]

            # L (List)에 추가
            L.append(sum)

        # answer (List)에 L(List) 추가
        answer.append(L)

    return answer

 

[Reference]

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

프로그래머스