Algorithm/Python
[Programmers]Lv 2.행렬의 곱셈
Pycoder
2019. 3. 20. 12:00
728x90
반응형
문제:
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 |
728x90
반응형