[Day25-3] 정사각형으로 만들기

문제

 

내 정답 코드

import Foundation

func solution(_ arr:[[Int]]) -> [[Int]] {
    var arr = arr
    let length: Int = arr.count > arr[0].count ? arr.count : arr[0].count
    
    if arr.count == length {
        for i in 0..<arr.count {
            (0..<length-arr[i].count).map {_ in arr[i].append(0) }
        }
    } else {
        (0..<length-arr.count).map {_ in arr.append(Array(repeating: 0, count: arr[0].count)) }
    }
    return arr
}

 

#

1. 1차원 배열의 길이와 2차원 배열의 길이를 가져와 더 큰 값을 length에 저장한다.

2. 1차원 배열의 길이가 부족하면 0을 채우고

3. 2차원 배열의 길이가 부족하면 0을 채운 배열을 채운다.

---

배운 기술