알고리즘 테스트/프로그래머스 문제풀이 및 해설

<프로그래머스 문제풀이: 행렬의 덧셈> Level 1 - 파이썬

개발린이 2020. 11. 27. 18:12


비교적 간단한 풀이였다:

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


하지만 이외에도 더 쉽게 푼 사람이 있었다


다른 사람의 풀이

프로그래머스에서 가장 많은 좋아요를 받은 풀이이다:

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


zip 함수를 사용하여 조금 더 편리하게 풀었다. zip함수에 대해서는 Python 공부하기에 올려놓도록 하겠다