오랜만에 접하는 행렬이라 당황했지만 행렬 곱셈 과정을 이해하면 삼중 for문으로 어렵지 않게 해결할 수 있다 !
수학 문제가 나오면 공식을 몰라서 헤매는 경우가 많은데 여러 문제를 풀면서 익숙해져야 할 것 같다.
- 문제 해결에 참고한 행렬의 곱셈
def solution(arr1, arr2):
answer = [[0]*len(arr2[0]) for _ in range(len(arr1))]
for i in range(len(arr1)):
for j in range(len(arr2[1])):
for k in range(len(arr1[1])):
answer[i][j] += arr1[i][k] * arr2[k][j]
return answer