[swift] 백준 - 11866번: 요세푸스 문제 0

11866번: 요세푸스 문제 0

문제 푸는 데 있어 도움이 되도록 나의 풀이와 개선된 풀이를 올립니다.
또한 풀이 후 다른 사람의 풀이를 보고 참고할만한 풀이도 올립니다.
- 문제에 따라 나의 풀이만 있을 수 있습니다.
- 해당 풀이들은 풀이 중 하나일 뿐 최선의 풀이는 아닐 수 있습니다.

 

---

 

문제 설명

 

---

 

나의 풀이

let input = readLine()!.split(separator: " ").map { Int($0)! }
var arr = Array((1...input[0]))
var result = [Int]()
var now = 0
while !arr.isEmpty {
    now = (now + input[1]-1) % arr.count
    result.append(arr[now])
    arr.remove(at: now)
}
print("<", terminator: "")
print("\(result[0])", terminator: "")
for i in result[1..<result.count] { print(", \(i)", terminator: "") }
print(">")

 

예제 문제 1을 예로 들어

 

입력 값 N만큼의 길이의 arr 배열을 만들고

(N이 7이라면 [1,2,3,4,5,6,7]을 만듭니다.)

 

배열을 순회하며

입력 값 K만큼 떨어진 거리의 값을 추출 및 제거했습니다.

 

K가 3이라면

N의 3번째 값

즉, 인덱스 0 + 2인

N[2]에 3을 추출하여

빈배열을 하나 생성하여 담아줍니다.

 

추출된 인덱스 즉 정수 2를 now변수에 저장하고

추출된 N[2]는 제거합니다.

 

while문은 arr배열이 비어질 때까지 반복합니다.

 

다시 기억해놨던 인덱스 N[2]부터 + 2만큼

N[4]를 추출 및 제거하게 됩니다.

 

배열을 무사히 순회하기 위해

% arr.count를 하여

배열 인덱스가 초과하면 0부터 다시 순회할 수 있게 했습니다.

 

---

 

개선된 풀이

print("<", terminator: "")
print("\(result[0])", terminator: "")
for i in result[1..<result.count] { print(", \(i)", terminator: "") }
print(">")

 

<3, 6, 2, 7, 5, 1, 4>

이런 출력 값을 뽑기 위해

위의 코드를 사용했었는데

 

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

 

해당 코드로 개선할 수 있었습니다.

 

---