✅파이썬 큐 문제(백준 2164번)

이상민·2023년 5월 19일
0

알고리즘

목록 보기
3/128

📖문제

  • N장의 카드가 있다. 각각의 카드는 차례로 1부터 N까지의 번호가 붙어 있으며, 1번 카드가 제일 위에, N번 카드가 제일 아래인 상태로 순서대로 카드가 놓여 있다.

  • 이제 다음과 같은 동작을 카드가 한 장 남을 때까지 반복하게 된다. 우선, 제일 위에 있는 카드를 바닥에 버린다. 그 다음, 제일 위에 있는 카드를 제일 아래에 있는 카드 밑으로 옮긴다.

  • 예를 들어 N=4인 경우를 생각해 보자. 카드는 제일 위에서부터 1234 의 순서로 놓여있다. 1을 버리면 234가 남는다. 여기서 2를 제일 아래로 옮기면 342가 된다. 3을 버리면 42가 되고, 4를 밑으로 옮기면 24가 된다. 마지막으로 2를 버리고 나면, 남는 카드는 4가 된다.

  • N이 주어졌을 때, 제일 마지막에 남게 되는 카드를 구하는 프로그램을 작성하시오.

📗입력

  • 첫째 줄에 정수 N(1 ≤ N ≤ 500,000)이 주어진다.

📘출력

  • 첫째 줄에 남게 되는 카드의 번호를 출력한다.

📄내가 작성한 코드

from collections import deque
import sys

N = int(sys.stdin.readline())
queue = deque()
add = []

for i in range(N):
    queue.append(i+1)

while len(queue) != 1:
    queue.popleft()
    add = queue.popleft()
    queue.append(add)
print(*queue)

👉코드 설명

  • 빠른 처리를 위해 queue 대신 from collections import deque를 통하여 deque를 사용하였다.
  • N장의 카드를 받기위해 deque()동작을 할 변수 queue를 선언하였다.
  • 임시저장을 위한 리스트 add를 선언하였다.
  • queue에 1부터 N까지 순차적으로 저장한다(for문은 0부터 시작하므로 i+1을 담는다)
  • 카드가 한장만 남을때까지 즉, queue의 길이가 1이 될때까지 while문을 동작시킨다.
    1. 가장 위쪽(리스트의 왼쪽) 숫자를 꺼낸다.
    1. 가장 위쪽 숫자를 한번더 꺼내서 add에 임시 저장한다.
    1. queue에 add를 저장한다.
  • queue 리스트의 요소값을 출력한다.

❗❗Key Point

  • 빠른 처리를 위해 deque()사용
  • 카드가 한장만 남을때까지 == queue 리스트 길이가 1
  • 위쪽 카드를 아래의 놓기 위해 pop()이후 임시저장을 해서 리스트에 append()
  • 출력시, 리스트 출력이 아닌 리스트의 요소값 출력

💢막혔던 부분

  • deque()사용을 위해 import하는 구문을 몰랐다.. from collection import deque 외우자.
  • print(queue)시 리스트가 출력이 되었는데, 요소값 출력하는 법을 몰랐다. 요소값 출력시 print(*queue)로 해야한다.
profile
개린이

0개의 댓글