프로그래머스 - 행렬의 곱셈(Level 2)/Wanna Be 컴잘알

문제 출처 - https://programmers.co.kr/learn/courses/30/lessons/12949

프로그래머스

-문제-

 

문제 설명

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

제한 조건

• 행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.
• 행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.
• 곱할 수 있는 배열만 주어집니다.

입출력 예

arr1arr2return

[[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]]

 

-설명-

 

행렬의 곱은

 

[a1,b1]                           [a1*c1 + b1*c2,  a1*d1+b1*d2]

[a2,b2]    x    [c1,d1]     =  [a2*c1 + b2*c2,  a2*d1+b2*d2]

[a3,b3]         [c2,d2]         [a3*c1 + b3*c2,  a3*d1+b3*d2]

 

3x2            2x2         

(행렬 곱셈 시, 앞에 행렬의 열의 수와 뒤에 행렬의 행의 수가 같아야하고 최종적으로 나오는 행렬은 앞에 행렬의 행의 수 x 뒤에 행렬의 열의 수 가된다.)

예를들면 5x3 과 3x7을 곱하면 결과 행렬은 5x7이 나온다.

 

answer이라는 2차원배열을 0으로 모두 초기화 시켜놓고 a1->b1->a2->b2->a3->b3 순서로 돌면서 해당 값을 x라고 지칭하면 x가 곱해져야할 수와 곱을 한뒤 answer += 곱한 값 을 누적시키는 방법을 취했다.

 

-코드-

def solution(arr1, arr2):
    b = len(arr1)
    a = len(arr2[0])

    answer = [[0 for col in range(a)] for row in range(b)]

    for i,value1 in enumerate(arr1): # 0 [1,2,3]
        for k,value2 in enumerate(value1): #1,2,3
                    for j in range(a):
                        answer[i][j] += arr1[i][k] * arr2[k][j]
    return answer