[백준] (Swift) 1158번 - 요세푸스 문제

 

문제 링크

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

1158번: 요세푸스 문제

 

내가 푼 풀이

• idx를 pop, push를 활용하여 구현하려고 했지만 저렇게 덧셈과 count를 활용한 나눗셈으로 규칙을 찾을 수 있다.
• 이 부분에서 조금 막혔지만, 규칙을 찾으니 완전히 이해가 갔다. 

import Foundation

let input = readLine()!.split(separator: " ").map { Int($0)! }
var nums: [Int] = []
var answer: [Int] = []
var idx = input[1] - 1

for i in 1..<input[0]+1 {
    nums.append(i)
}

while true {
    answer.append(nums.remove(at: idx))
    if nums.isEmpty { break }
    idx = (idx + input[1]-1) % nums.count
}


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