프로그래머스 / 구현 / 행렬의 곱셈

문제

---

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

프로그래머스

 

 

입출력

---

 

해설

---

행렬에서의 곱셈을 그대로 구현하는 문제이다.

 

Arr1, Arr2를 통해서 Arr3를 만드는 문제이고, Arr3를 만드는 이론은 다음과 같다.

 

Arr3[i][j] := Arr1[i][1] * Arr2[1][j] + ... Arr1[i][n] * Arr2[n][j]

 

문제 그대로 3중 반복문을 통해서 문제를 풀이하였다.

Arr3의 값을 넣으려는 2중 포문을 열고, Arr1과 Arr2 곱의 합을 구하는 for문을 이용하여 풀이하면 된다.

 

정답코드

---

'''
'25. 08. 31

1. arr1, arr2의 곱을 반환하는 함수를 완성하여라.

arr1 3, 2
arr2 2, 2

rst 3, 2

rst[i][j] := sum(arr1[i][:] * arr2[:][j])이다.
'''
def solution(arr1, arr2):
    new_row = len(arr1)
    new_col = len(arr2[0])
    answer = [[0]*new_col for _ in range(new_row)]
    
    #10^4
    for i in range(new_row):
        for j in range(new_col):
            v = 0
            for col in range(len(arr1[0])):
                    v += arr1[i][col] * arr2[col][j]
            answer[i][j] = v
    return answer

 

깃허브 : https://buly.kr/9tBJVH6

유튜브 : https://youtu.be/Kfy1WHMLgds?si=soibNoH0ZZuXV246