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

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

코딩테스트 연습 - 행렬의 덧셈

풀이

 

행렬의 덧셈 방법과 2차원 배열을 어렵지 않게 다룰 수 있다면 전혀 어렵지 않다.

우선 덧셈 결과를 저장할 2차원 배열 result로서 입력으로 주어진 배열과 같은 크기를 가진 배열을 선언한다.

그 후 이중 for loop를 통해 arr1과 arr2 2차원 배열(행렬)에서 같은 인덱스의 값끼리 더하여 result 배열에 저장하면 된다.

 

끝!

func solution(_ arr1:[[Int]], _ arr2:[[Int]]) -> [[Int]] {
    var result: [[Int]] = Array(repeating: Array(repeating: 0, count: arr1[0].count), count: arr1.count)
    for i in 0..<arr1.count {
        for j in 0..<arr1[0].count {
            result[i][j] = arr1[i][j] + arr2[i][j]
        }
    }
    
    return result
}