새소식

⌨️ Algorithms/프로그래머스

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

2022. 8. 5. 20:50

  • -

https://school.programmers.co.kr/learn/courses/30/lessons/12950

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

"""
입출력 예시)

arr1 = [[1,2],[2,3]], arr2 = [[3,4],[5,6]] -> [[4,6],[7,9]]
arr1 = [[1],[2]], arr2 = [[3],[4]] -> [[4],[6]]
"""

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

        answer.append(arr_sum) # answer[i][j] = arr1[i][j] + arr2[i][j]
            
    return answer

 

  • 두 행렬은 행과 열의 크기가 같아야 함
    • -> i == j
Contents

포스팅 주소를 복사했습니다

이 글이 도움이 되었다면 공감 부탁드립니다!