[프로그래머스 Lv2] 행렬의 곱셈(python)

이진규·2022년 1월 16일
1

프로그래머스(PYTHON)

목록 보기
21/64

문제

https://programmers.co.kr/learn/courses/30/lessons/12949

나의 코드

"""
1. 아이디어
그림그려서 직접 풀어봐야한다. 어지럽다 ;;

2. 시간복잡도
O(N^3)
"""

def solution(arr1, arr2):
    
    answer = []
    
    for i in range(len(arr1)):
        tmp = []
        
        for j in range(len(arr2[0])):
            sum = 0
            
            for k in range(len(arr2[0])):
                sum += (arr1[i][k] * arr2[k][j])
            
            tmp.append(sum)
            
        answer.append(tmp)
    
    return answer
    

느낀점

그림 그려봐도 헷갈리는 문제라 다시 풀어봐야 할 필요가 있다.

profile
항상 궁금해하고 공부하고 기록하자.

0개의 댓글

관련 채용 정보