행렬의 덧셈

문제

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

프로그래머스

 

내 답안 / 다른 사람들의 답안

func solution(_ arr1:[[Int]], _ arr2:[[Int]]) -> [[Int]] {
    var result = Array(repeating: Array(repeating: 0, count: arr1[0].count), count: arr1.count)

    for i in 0..<arr1.count {
        for j in 0..<arr1[i].count {
            result[i][j] = arr1[i][j] + arr2[i][j]
        }
    }

    return result
}

- 각 행과 열의 항목을 더한 배열을 반환하는 방법을 사용했다.

 

// 풀이 1
import Foundation

func solution(_ arr1:[[Int]], _ arr2:[[Int]]) -> [[Int]] {
    return zip(arr1, arr2).map{zip($0,$1).map{$0+$1}}
}

// 풀이 2
func solution(_ arr1:[[Int]], _ arr2:[[Int]]) -> [[Int]] {
    var arr: [[Int]] = []

    for n in 0..<arr1.count {
        var temp:[Int] = []
        for m in 0..<arr1[n].count {
            temp.append(arr1[n][m] + arr2[n][m])
        }
        arr.append(temp)
    }

    return arr
}

- zip을 사용해보려 했지만 단순히 배열을 합치는 정도여서 반복문을 사용했는데, 풀이 1과 같은 방법으로 사용하면 됨을 알 수 있었다.

 

점수: +1