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 answerps/problems/programmers/12949.txt · 마지막으로 수정됨: 2021/07/08 15:50 저자 teferi
토론