[programmers] 프로그래머스 행렬의 덧셈(Python)

(주)그렙

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

(Python)

(글쓴날 : 2020.06.15)

 

---

* programmers, 프로그래머스 문제 Python 언어 풀이입니다.

* 소스 코드의 저작권은 글쓴이에게 있습니다.

---

 

 

프로그래머스 행렬의 덧셈

---

1) 문제

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

코딩테스트 연습 - 행렬의 덧셈

---

2) 풀이 과정

* 시간 복잡도 : O(n²)

 

크기가 서로 같은 행렬 2개가 주어질 때, 두 행렬의 덧셈 결과를 반환하는 문제입니다.

 

저의 경우, Python을 사용했습니다.

먼저, 주어지는 행렬의 크기만큼 0으로 초기화한 2차원 리스트를 만든 뒤, 2중 반복문을 돌려 해당 리스트에 두 행렬의 합을 저장하여 문제를 해결했습니다.

---

3) 코드

 

* Python 코드

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

---