[백준] 요세푸스 문제0 11866 (파이썬)

스케치

코드

• 기본적으로 큐를 활용해서 문제를 풀면 되지만 K번째 값을 뽑는 도중 Q가 K보다 작아지게 될 때를 유의하여 처리하여야 한다.
• Q < K 작은 구간은 나눗셈으로 인덱스를 뽑아서 구현하였다.
• 출력양식 공부가 필요해보인다. 
• 2021.03.18 - [Programming/Python] - [파이썬] print() 원하는 양식으로 출력하기 (백준 11866 요세푸스 문제)

[파이썬] print() 원하는 양식으로 출력하기 (백준 11866 요세푸스 문제)

# 요세푸스 문제0
from collections import deque

n,k = map(int,input().split())

q = deque()
arr = []
for i in range(1,n+1):
     q.append(i)
while q: # 1 2 1
    if len(q) >= k:
        cnt = 1
        while cnt < k:
            q.append(q.popleft())
            cnt += 1
        arr.append(q.popleft())
    else:
        index = k % len(q) 
        cnt = 1
        if index == 0:
            arr.append(q.pop())
        else:
            while cnt < index:
                q.append(q.popleft())
                cnt += 1
            arr.append(q.popleft())

# 출력형식에 맞게 출력하는 것이 쉽지 않았다...
print('<' ,end='')
for i in range(len(arr)):
    if i != len(arr) - 1:
        print("%d, " %arr[i], end='')
    else:
        print("%d>" %arr[i])