#19 알고리즘연습 행렬의덧셈 python

행렬의 덧셈

문제 설명

행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요.

제한 조건

• 행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.

입출력 예

arr1	arr2	return
[[1,2],[2,3]]	[[3,4],[5,6]]	[[4,6],[7,9]]
[[1],[2]]	[[3],[4]]	[[4],[6]]

---

---

문제풀이 IDEA

행렬문제를 풀때는 중첩 for문을 자주 사용해서 중첩 for을 이용한 list append를 이용해
각 row별로 연산을 수행해줬다.

---

---

나의 코드

1 2 3 4 5 6 7 8 9

def solution(arr1, arr2): answer = [] for i in range(0,len(arr1)): #range(len(arr1))으로 바꿔줘도 된다 row_list = [] #answer에 append해줄 row를 보관하는 임시 row_list for j in range(0,len(arr1[0])): #row안에 있는 col을 하나하나 지칭하기위해 row_list.append(arr1[i][j] + arr2[i][j]) #row별로 차곡차곡 col의 합을 append해준다 answer.append(row_list) #row_list의 작성이 완료됭ㅆ으면 asnwer에 추가 해준다 return answer

다른코드

1 2 3 4 5 6 7 8 9 10

def sumMatrix(A,B): answer = [[A[i][j] + B[i][j] for j in range(len(A[0]))] for i in range(len(A))] return answer 위의 코드에서 list comprehension을 사용하여 문제를 풀었는데 중첩 for문을 list comprehension에 넣는다는 생각이 참신했다. 그리고 나는 더한 원소들을 list에 담아야 list안에 추가해줄수 있다 생각했는데 아예 하나하나의 원소를 더해서 그 원소를 []로 묶어주고 그 list를 다시 for 문으로 list를 생성해 주는 아이디어를 배웠다 예를 들면 A[i][j] = A[i][j] + B[i][j] 이런 아이디어도 있는걸 생각할수 있어야 위와같은 아이디어도 나올수 있을것 같다. ㅇㅇㅇ

출처 : 프로그래머스