[swift] 프로그래머스 - 정사각형으로 만들기

프로그래머스

---

성공 풀이

import Foundation

func solution(_ arr:[[Int]]) -> [[Int]] {
    var h = arr.count
    var y = arr[0].count
    var n = h-y
    if n > 0 {
        for i in 0..<arr.count {
            return arr.map {$0+Array(repeating: 0, count: n)} 
        }
    } else if n < 0 {
        return arr + Array(repeating: Array(repeating: 0, count: arr[0].count), count: -n)
    }
    return arr
}

---