[프로그래머스]행렬의 덧셈

문제 링크 : https://school.programmers.co.kr/learn/courses/30/lessons/12950

프로그래머스

본 문제를 간략히 설명하면 다음과 같다.

열과 행의 크기가 같은 두 개의 행렬이 주어질 때, 두 행렬을 더한 행렬을 반환하는 것이다.

 

코드를 작성하기 위한 순서는 다음과 같다.

1. 두 행렬의 열과 행의 크기가 같으므로 아무 행렬의 크기를 첫번째 반복문의 조건으로 잡아준다.

2. 행렬을 더하는 순서는 각 열의 행을 다 더하고 다음 열로 넘어가기 때문에, 행렬의 한 열에 몇 개의 행이 들어있는가를 두번째 반복문의 조건으로 잡아준다.

3. 더한 값을 임의의 리스트에 넣어주고 해당 리스트를 answer에 넣어준다.

def solution(arr1, arr2):
    answer = []
    for i in range(len(arr1)): #1번
        temp=[]
        for j in range(len(arr1[i])): #2번
            temp.append(arr1[i][j]+arr2[i][j]) #3번
        answer.append(temp) #3번 : 임의의 리스트에 넣고 answer에 넣어야 [[1,2],[3,4]]와 같은 행렬의 모습을 만들 수 있다.
    return answer