[백준][1158] 요세푸스 문제 | 파이썬

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

 

👉 문제 분석

스택과 큐의 구조 사용

 

👉 알고리즘 설계

양의 정수 K를 입력받고, K 번째 순서(주기)마다 사람을 제거한다.

제거된 사람을 순서대로 출력한다.

 

👉 코드

import sys
from collections import deque

people = deque()

n, k = map(int, sys.stdin.readline().split())

for i in range(1, n+1):
    people.append(i)

result = []
i = 0
while len(people) > 0:
    for j in range(1, k):
        people.append(people.popleft())
    result.append(people.popleft)

print(str(result).replace('[', '<').replace(']', '>'))

 

먼저 순열 n들을 입력받아 people에 집어넣는다. 

while문에서 people에 숫자가 있는동안 계속 돌아가게 된다.

for문에서는 처음부터 k-1까지 제일 앞 요소가 people에 들어가게 된다 (원을 이루며 앉아있기 때문이다.)

result 리스트를 만들어 people의 맨 앞요소(k 주기로 삭제될 요소)를 popleft를 통해 삭제하고 append해준다.

이제 result를 프린트해주면 되는데 수가 < > 안에 들어있으니, 리스트를 프린트하되 [ ] 부분을 < >으로 각각 replace() 를 통해 대체해주면 된다.  

 

👉 틀린 이유

1. TypeError: 'str' object cannot be interpreted as an integer
2. 출력 부분에서 < > 부분 때문에 리스트로 출력해야하나 고민했다.
3. 런타임 메러

 

👉  틀린 부분 수정 or 다른 풀이

1. map() 사용으로 해결
2. replace() 사용으로 해결
3. popleft()사용으로 해결

👉 느낀점 or 기억할정보

1. map은 리스트의 요소를 지정된 함수로 처리해주는 함수이다.
2. 리스트를 출력할때 같이 출력되는 문자([ ])를 replace로 문자 대체를 할 수 있다.
3. popleft()는 제일 앞의 요소가 삭제된다.