0903 | 알고리즘 오답노트

왜 알고리즘 오답노트?

끝까지 풀었는데 시간 초과등의 문제가 생기면 의욕저하가 심각해서 해결을 안 보고 넘어갈 때가 있다. 이건 끈기와 집중력과 프로페셔널리즘의 문제라고 생각해서 오답노트를 시작하게 되었다.

문제설명 📖

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

내가 푼 풀이

def solution(arr1, arr2):
    return [[sum([arr1[a][y]*arr2[y][z] for y in range(len(arr2[0]))]) for z in range(len(arr2))] for a in range(len(arr1))]