행렬의 덧셈은 크게 2가지 방식으로 풀이할 수있다.
바로 numpy 라이브러리를 활용하는 방식과 행렬의 덧셈구조를 그대로 구현하는 방식이다.
아래 방식은 numpy 라이브러리를 이용하는 방식이다.
import numpy as np
def solution(arr1, arr2):
arr1 = np.array(arr1)
arr2 = np.array(arr2)
answer = arr1+arr2
return answer.tolist()
이제 행렬의 덧셈을 그대로 구현한 방식을 알아보자.
행렬의 덧셈은 행렬의 각 행,열에 해당하는 위치를 그대로 유지한채 각 값을 더해주면 된다.
def solution(arr1, arr2):
answer = []
for i in range(len(arr1)):
temp=[]
for j in range(len(arr1[i])):
temp.append(arr1[i][j]+arr2[i][j])
answer.append(temp)
return answer
'알고리즘 & 코딩테스트 > 코딩테스트' 카테고리의 다른 글
Programmers Lv. 1 "최대공약수와 최소공배수" (0) | 2023.05.22 |
---|---|
Programmers Lv. 1 "직사각형 별찍기" (0) | 2023.05.22 |
Programmers Lv.1 "문자열 다루기 기본" (0) | 2023.05.22 |
Programmers Lv.1 "부족한 금액 계산하기" (0) | 2023.05.22 |
Programmers Lv.1 "약수의 개수와 덧셈" (0) | 2023.05.16 |