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

SEUNGCHAN BAEK·2021년 3월 12일
0

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

다른사람 풀이 참고

def solution2(arr1,arr2):
    total=[]
    answer=[]
    for a,b in zip(arr1,arr2):
        for c,d in zip(a,b):
            total.append(c+d)
        answer.append(total)
        total=[]
    return answer

zip이라는 개념을 생각하지 못했는데 이 문제를 풀면서 zip에 대해서 한번 더 공부하게 되었다.

def solution3(arr1, arr2):
    return [[c + d for c, d in zip(a, b)] for a, b in zip(arr1,arr2)]

이건 위에 zip 코드를 한 줄 코드로 만들었을때의 코드이다
존경스럽다....

profile
백엔드 개발자가 되는 그날까지

0개의 댓글

관련 채용 정보