[Swift] 프로그래머스 n^2 배열 자르기(lv. 2)

[Swift] 프로그래머스 n^2 배열 자르기(lv. 2)

 

 

1. 문제

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

프로그래머스

2. 접근

 

 

 

3. 코드

import Foundation

func solution(_ n:Int, _ left:Int64, _ right:Int64) -> [Int] {
    var array = [Int]()
    var line = left / Int64(n)
    var row = left % Int64(n)
    
    var k: Int64 = {
        if line < row {
            return row + 1
        } else {
            return line + 1
        }
    }()
    while array.count != right - left + 1 {
        array.append(Int(k))
        if line <= row {
            k += 1
        }
        row += 1
        if row == n {
            row = 0
            line += 1
            k = line + 1
        }
    }
    return array
}