[프로그래머스 Lv.2] 행렬의 곱셈 (Python)

지윤·2023년 5월 29일
0

👩🏻‍💻

오랜만에 접하는 행렬이라 당황했지만 행렬 곱셈 과정을 이해하면 삼중 for문으로 어렵지 않게 해결할 수 있다 !
수학 문제가 나오면 공식을 몰라서 헤매는 경우가 많은데 여러 문제를 풀면서 익숙해져야 할 것 같다.

  • 문제 해결에 참고한 행렬의 곱셈

코드

def solution(arr1, arr2):
    answer = [[0]*len(arr2[0]) for _ in range(len(arr1))]
    for i in range(len(arr1)):
        for j in range(len(arr2[1])):
            for k in range(len(arr1[1])):
                answer[i][j] += arr1[i][k] * arr2[k][j]
    return answer
profile
떠돌이 컴공

0개의 댓글