[프로그래머스] '행렬의 곱셈' 알고리즘 풀이 - Python

Contents

1. 문제 설명
   
   [제한사항]
   
   [입출력 예]
2. 알고리즘 분석 
   
   [나의 풀이]
   
   [Most 1 의 풀이]

---

 

문제 설명

 

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

 

 

제한사항

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

 

입출력 예

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

 

 

알고리즘 분석

 

• 나의 풀이

def solution(arr1, arr2):
    answer = [[0] * len(arr2[0]) for i in range(len(arr1))]
    # arr2를 곱하기 쉽도록 시계 방향으로 90도 돌림 
    turn_arr2 = [[arr2[i][j] for i in range(len(arr2))] for j in range(len(arr2[0]))]

    for i in range(len(arr1)):
        for j in range(len(turn_arr2)):
            arg = 0  # answer[i][j]에 들어갈 값
            for x, y in zip(arr1[i], turn_arr2[j]): # 같은 인덱스의 값끼리 매칭해줌
                arg += x * y # 값들을 곱해서 arg에 더해줌 
            answer[i][j] = arg 

    print(answer)
    return answer

 

• 나의 풀이 ( numpy Ver )

import numpy as np
def solution(arr1, arr2):
    arr1, arr2 = np.array(arr1), np.array(arr2) # list를 array로 형변환 
    answer = np.dot(arr1, arr2).tolist() # dot : 행렬곱, tolist : array를 list형으로 형변환 
    print(answer)
    return answer

 

• Most1의 풀이 

# Most 1의 풀이 원본 
def solution(A, B):
    return [[sum(a*b for a, b in zip(A_row,B_col)) for B_col in zip(*B)] for A_row in A]
    
# 한 줄이라 보기 어려워서 풀어봤습니다. 
# 대충 이렇습니다. 
def solution(arr1, arr2):
    answer = [[] for i in range(len(arr1))]
    for i, a in enumerate(arr1):
        for b in zip(*arr2):
            arg = 0
            for x, y in zip(a, b):
                arg += x * y
            answer[i] += [arg]
    return answer

 

 

 

코딩테스트 연습 - 행렬의 곱셈