11866번 요세푸스 문제 : SWIFT

https://www.acmicpc.net/problem/11866

11866번: 요세푸스 문제 0

 

 

import Foundation

let n = readLine()!.split(separator: " ").map{Int(String($0))!}

let N = n[0]
var K = n[1]

var arr = Array(1...N)
var killed = [Int]()
var cnt = K // 제거

while !arr.isEmpty {
    if cnt <= arr.count {
        killed.append(arr[cnt-1])
        arr.remove(at: cnt-1)
        cnt = cnt + K - 1
    }else {
        cnt -= arr.count
    }
}

print("<" + killed.map({String($0)}).joined(separator: ", ") + ">")