사용자 도구

사이트 도구


ps:problems:programmers:12949

행렬의 곱셈

ps
링크programmers.co.kr/…
출처프로그래머스
문제 번호12949
문제명행렬의 곱셈
레벨Level 2
분류

기초

시간복잡도O(nmk)
인풋사이즈n<=100, m<=100, k<=100
사용한 언어Python
해결날짜2021/07/08

풀이

  • 그냥 기초적인 행렬 곱셈 문제.
  • 이론적으로는 슈트라센 알고리즘등을 사용하면 좀더 시간복잡도를 줄일수 있지만 (O(n^3)을 O(n^2.8)정도로..), n이 100정도의 범위에서는 별 의미가 없을 듯 해서 그냥 단순한 방법으로 구현.

코드

"""Solution code for "Programmers 12949. 행렬의 곱셈".

- Problem link: https://programmers.co.kr/learn/courses/30/lessons/12949
- Solution link: http://www.teferi.net/ps/problems/programmers/12949
"""


def solution(arr1, arr2):
    answer = []
    for arr1_row in arr1:
        answer.append([(sum(x * y for x, y in zip(arr1_row, arr2_col))) 
                       for arr2_col in zip(*arr2)])
    return answer

토론

댓글을 입력하세요:
H F U Y B
 
ps/problems/programmers/12949.txt · 마지막으로 수정됨: 2021/07/08 15:50 저자 teferi