[프로그래머스] 행렬의 곱셈 / 파이썬 / Python

 

 

💡solutions )

💬 예를 들어, AXB 행렬과 BXC 행렬의 곱셈 -> 두 행렬에서 B의 자리가 같은 숫자일 때만 곱셈이 가능하며 -> 곱셈의 결과는 AXC행렬이 된다.

💬 아래 코드에서 answer은 곱셈의 결과가 되는 AXC행렬이고,

💬 i가 A / j가 C / k가 B로 삼중 반복문으로 행렬의 곱셈을 구현했다.

 

🎫code )

def solution(arr1, arr2):
    r = len(arr1)
    c = len(arr2[0])
    answer = [[0]*c for _ in range(r)]
    
    for i in range(r):
        for j in range(c):
            for k in range(len(arr1[0])):
                answer[i][j] += arr1[i][k]*arr2[k][j]
    return answer

 

📌 description )

문제출처 : programmers.co.kr/learn/courses/30/lessons/12949?language=python3

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

문제 설명 2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱한 결과를 반환하는 함수, solution을 완성해주세요. 제한 조건 행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다. 행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다. 곱할 수 있는 배열만 주어집니다. 입출력 예