정사각형으로 만들기

문제

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

프로그래머스

 

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

import Foundation

func solution(_ arr:[[Int]]) -> [[Int]] {
    var arr = arr
    let colCount = arr[0].count
    let rowCount = arr.count

    for index in arr.indices {
        if arr[index].count < rowCount {
            arr[index].append(contentsOf: Array(repeating: 0, count: rowCount - arr[index].count))
        }
    }

    if rowCount < colCount {
        arr.append(contentsOf: Array(repeating: Array(repeating: 0, count: colCount), count: colCount - rowCount))
    }

    return arr
}

- 처음 for문에서는 각 요소의 개수가(열의 개수가) 행의 개수와 맞는지 체크했다. 열이 행보다 적다면 부족한 만큼 0을 append 해줬다.

- 그렇게 이미 있는 행의 열 개수를 맞춰준 다음, 행의 개수가 열 개수와 맞는지 체크하여 적다면 열 개수만큼의 0이 들은 배열을 부족한 행의 개수만큼 append 해줬다.

 

// 풀이 1
import Foundation

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

// 풀이 2
import Foundation

func solution(_ arr:[[Int]]) -> [[Int]] {
    let rowCount = arr.count
    let columnCount = arr[0].count
    if rowCount > columnCount {
        return arr.reduce(into: [[Int]]()) {
            $0.append($1 + Array(repeating: 0, count: rowCount - columnCount))
        }
    } else if columnCount > rowCount {
        return (1...columnCount-rowCount).reduce(into: arr) { arr, _ in
            arr.append(Array(repeating: 0, count: columnCount))
        }
    }
    return arr
}

 

 

점수: +4