알고리즘/알고리즘 문제 풀이
[프로그래머스 Lv.2] 행렬의 곱셈 (Python)
용꿀
2024. 5. 10. 09:47
문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/12949
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
풀이
행렬곱을 구현하면 되는 단순한 문제이다. 행렬곱의 경우 아래와 같은 방식으로 계산한다.
이런 계산 방식을 코드로 구현하면 된다.
def solution(arr1, arr2):
answer = []
for m in range(len(arr1)): # 피연산자(arr1)의 각 행 별로 계산
arr = []
for r in range(len(arr2[0])): # 연산자(arr2)의 열의 수만큼 반복
sum = 0
for n in range(len(arr2)): # 피연산자(arr1)의 열의 수 == 연산자(arr2)의 행의 수만큼 반복
sum += arr1[m][n]* arr2[n][r]
arr.append(sum)
answer.append(arr)
return answer