🐍python3/Programmers

프로그래머스[Lv. 1]_행렬의 덧셈

박필중 2023. 6. 7. 16:19
728x90

🤷‍♂️문제설명

행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다.

2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요.

 

🙅‍♀️제한조건

  • 행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.

 

💁입출력 예

arr1 arr2 return
[[1,2],[2,3]] [[3,4],[5,6]] [[4,6],[7,9]]
[[1],[2]] [[3],[4]] [[4],[6]]

 

🙆코드

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

 

더보기

🗨️첨언

2차원 배열을 다루는 법은 for문 안에 또 for문을 만드는 것이다.

그리고 이전에 말했던 것처럼 필요한 변수, 리스트가 일시적으로 필요하다면  for문 속에 만들면 된다.

반복할 때마다 생성하고 초기화를 하기 때문에 순서를 잘 맞춰서 작성해야 한다.

 

for 문

   필요한 변수 혹은 리스트

      for 문

  변수를 넣는 코드

for 문
	필요한 변수 혹은 리스트
	for 문
	변수를 넣는 코드

 

728x90