[Coding Test] 행렬의 덧셈

프로그래머스 코딩테스트 1단계 : 행렬의 덧셈

문제 설명

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

문제 풀이

• 이중 for문을 이용해 각각의 자리의 수를 더함
• arr1에 arr2를 더하여 arr1을 출력

정답 코드

def solution(arr1, arr2):

    for i in range(len(arr1)):
        for j in range(len(arr1[0])):
            arr1[i][j] += arr2[i][j]

    return arr1

다른 사람 코드

def sumMatrix(A,B):
    answer = [[c + d for c, d in zip(a,b)] for a, b in zip(A,B)]
    return answer

• zip 함수 사용
  [기초문법] zip 함수

* 이 문제는 프로그래머스 코딩테스트 연습 1단계 문제입니다.