코딩테스트

프로그래머스 - 행렬의 덧셈 (레벨1, swift)

momo_9 2020. 6. 25. 22:19

 

문제출처 👇

programmers.co.kr/learn/courses/30/lessons/12950

 

문제설명 👀

행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 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]]

 

 

문제풀이

-> 행렬의 같은 위치의 숫자들 끼리 더하면 되는 문제이다.

2중 반복문을 사용해서 해결했다.

 

1) 행렬의 행만큼 반복문을 돌린다. (arr1.count)

 -> 이때 각 행을 빈값으로 초기화 해준다 (result.append([])) << 이 부분을 놓쳐서, 살짝 헤매었다.

2) 그 반복문 속에서 해당 행의 열만큼 또 반복문을 돌린다 (arr1[i].count)

3) arr1과 arr2의 같은 행, 같은 열끼리 더해준다 (result[i].append(arr1[i][j] + arr2[i][j]))

 

아래는 위 내용을 구현한 전체 코드이다.