문제:
2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱합 결과를 반환하는 함수, solution을 완성해주세요,
풀이 방법:
행렬의 곱셈을 구하는 방법은 위와 같다. 이를 구현하기 위해서 미리 0행렬인 C를 만들어두고 곱셈값들을 더하는 방식으로 구현하였다.
1 2 3 4 5 6 7 8 9 10 11 12 | def solution(arr1, arr2): C=[] for m in range(len(arr1)): C.append([0,0]) while len(C[m]) < len(arr2[0]): C[m].append(0) for i in range(len(arr1)): for j in range(len(arr2[0])): for k in range(len(arr1[0])): C[i][j]+=arr1[i][k]*arr2[k][j] return C | cs |
'Algorithm > Python' 카테고리의 다른 글
| [Programmers]Lv 2.JadenCase 문자열 만들기 (0) | 2019.03.22 |
|---|---|
| [Programmers]Lv 3.짝지어 제거하기 (0) | 2019.03.21 |
| [Programmers]Lv 3.최고의 집합 (0) | 2019.03.19 |
| [Programmers]Lv 2. 피보나치 수 (0) | 2019.03.18 |
| [Programmers]Lv 3.줄 서는 방법 (0) | 2019.03.17 |