[ 알고리즘 ] 백준 1158 - 요세푸스 문제
문제 링크 : link 티어: 실버4 1158번: 요세푸스 문제 첫째 줄에 N과 K가 빈 칸을 사이에 두고 순서대로 주어진다. (1 ≤ K ≤ N ≤ 5,000) www.acmicpc.net 아이디어 코드 from collections import deque def main(): N, K = map(int, input().split(' ')) queue = deque([(i + 1) for i in range(N)]) ans = [] for i in range(N): queue.rotate(-(K - 1)) ans.append(str(queue.popleft())) ans = ', '.join(ans) print(f'') if __name__ == '__main__': main()
CS/알고리즘
2022. 9. 21. 01:10