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

문제 설명

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

제한 조건

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

입출력 예

arr1arr2return

[[1,2],[2,3]]	[[3,4],[5,6]]	[[4,6],[7,9]]
[[1],[2]]	[[3],[4]]	[[4],[6]]

 

 

단순한 문제인 것 같은데..... 나는 좀 풀면서 고민했다.

처음에는 단순히 answer[i][j]에 더한걸 대입하려고 했는데,

list out of range index..? 오류가 났다.

answer = [[]] 로 초기화 시킨 2차원 배열에 [i][j]를 찾아서 넣으라고 하니... 빈 리스트인데 그 index가 어디 있냐고 찾는건 당연하다

그래서 0으로 모두 초기화 시키고 대입했다.

 

def solution(arr1, arr2):
    
    n = len(arr1)
    m = len(arr1[0])
    
    answer = [list([0]*m) for _ in range(n)]
    
    for i in range(len(arr1)):
        for j in range((len(arr1[0]))):
            temp=0
            temp = arr1[i][j]+arr2[i][j] 
            answer[i][j] = temp
    
    return answer