본문 바로가기

알고리즘 & 코딩테스트/코딩테스트

Programmers Lv.1 "행렬의 덧셈"

Programmers Lv.1 행렬의 덧셈, 연습문제, 정답률 79%

행렬의 덧셈은 크게 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