프로그래머스 - 행렬의 덧셈 / Swift

문제 링크 : https://programmers.co.kr/learn/courses/30/lessons/12950

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

 

풀이

임시 배열 tmp를 만들어 각 행의 덧셈 결과를 삽입한다.

한 행이 끝나면 정답 배열 ans에 tmp를 삽입하고 다음 행으로 넘어간다.

모든 행렬의 덧셈이 완료되면 리턴한다.

 

 

코드

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