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

김멉덥·2023년 8월 5일
0

알고리즘 공부

목록 보기
81/171
post-thumbnail
post-custom-banner

문제

프로그래머스 연습문제


코드 구현

def solution(arr1, arr2):
    answer = []

    mul = 0
    sub = []

    for n in arr1:
        for i in range(len(arr2[0])):
            for j in range(len(n)):
                mul += n[j] * arr2[j][i]
            sub.append(mul)
            mul = 0
        answer.append(sub)
        sub = []

    return answer

풀이

  • A x B 행렬과 C x D 행렬은 B와 C의 길이가 같아야 한다!!
    → 처음에 for문의 range를 거꾸로 설정해주어서 런타임 에러가 발생하였다.
  • 참고한 테스트케이스 : https://school.programmers.co.kr/questions/7976

profile
데굴데굴 뚝딱뚝딱 개발기록
post-custom-banner

0개의 댓글