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

손주애·2020년 12월 17일
0

코딩테스트

목록 보기
9/22
post-custom-banner

def solution(arr1, arr2):
    answer = [[0 for _ in range(len(arr2[0]))] for _ in range(len(arr1))]

    for i in range(len(arr1)):
        for j in range(len(arr2[0])):
            k = 0
            temp = 0
            while k < len(arr1[0]):
                temp += arr1[i][k] * arr2[k][j]
                k += 1

            answer[i][j] = temp

    return answer

profile
백엔드 개발자입니다:)

0개의 댓글