본문 바로가기

알고리즘/test

python) 프로그래머스 - 행렬의 덧셈

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

 

프로그래머스

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

programmers.co.kr

def solution(arr1, arr2):
    # 행렬의 행과 열 수 가져오기
    rows, cols = len(arr1), len(arr1[0])
    
    # 결과를 보관할 새 행렬을 만듭니다.
    result = [[0 for j in range(cols)] for i in range(rows)]
    print(result)
    
    # 행렬의 행과 열을 반복하고 해당 요소를 추가합니다.
    for i in range(rows):
        for j in range(cols):
            result[i][j] = arr1[i][j] + arr2[i][j]
    
    # 결과 행렬 반환
    return result

풀이

rows 와 cols에 행렬의 수를 가져옵니다.

result라는 결과를 보관할 행,열,의 수가 같은 빈 행렬을 만들어줍니다.

행과 열만큼 for문을 사용하여서 result에 각 요소의 합을 넣어줍니다.