[프로그래머스] - 행렬의 곱셈

문제

---

[프로그래머스] - 행렬의 곱셈

 

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

프로그래머스

 

문제 이해

---

문제 설명

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

제한 조건

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

List 끼리의 곱이 아닌, 행렬의 곱으로 계산이 되어야 합니다.

=> arr1의 row( N*M)과 arr2(M*T)의 Column 곱의 합(N*T)

 

알고리즘

---

1. N,M,T를 활용한 3중 포문으로 구현 

2. 혹은 Zip 함수를 활용하여 Column을 나타내어 계산

 

코드

---

# 알고리즘1
def solution(first_arr, second_arr):
    answer = []
    n=len(second_arr)
    for t in range(len(first_arr)):
        temp=[]
        for i in range(len(second_arr[0])):
            temp_value=0
            for j in range(len(first_arr[0])):
                temp_value+=first_arr[t][j]*second_arr[j][i]

            temp.append(temp_value)
        answer.append(temp)
    return answer

# 알고리즘2
# Programmers-행렬의 곱셈
def solution(A, B):
    answer = []
    for a_row in A:
        temp = []
        for b_col in zip(*B):
            #print( "a_row : {} ".format(a_row))
            #print( "b_col : {} ".format(b_col))
            temp.append(sum([ a*b for a,b in zip(a_row,b_col) ]))
        answer.append(temp)

    return answer

solution([[2, 3, 2], [4, 2, 4], [3, 1, 4]], [[5, 4, 3], [2, 4, 1], [3, 1, 1]])