[백준/실버4] 2164번 - 카드2 (Python)

🔎 문제 링크

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

2164번: 카드2

 

✏️ 문제 설명

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

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

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

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

 

⌨️ 입력

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

#입력 예시
6

 

🖥️ 출력

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

#출력 예시
4

 

🔦 문제 풀이

해당 문제는 큐 알고리즘을 사용해 풀 수 있는 문제입니다. 문제 설명을 보면 크게 2가지 행동을 파악할 수 있습니다. 가장 맨 위에 있던 카드는 버리고, 그다음 카드는 카드의 맨 아래로 옮겨집니다. 이런 행동을 마지막 카드 하나만 남을 때까지 반복하고, 최종적으론 마지막 카드를 출력합니다.

 

이런 과정을 조금 다르게 해석하면, 홀짝으로 나누어 볼 수 있습니다. 자세히 설명하자면, 홀의 경우 가장 위의 카드를 버리고, 짝의 경우 가장 위의 카드를 맨 뒤로 옮긴다로 해석해 볼 수 있습니다. 여기서 맨 뒤로 옮기는 과정이 큐의 특징 중 하나인 FIFO입니다. 따라서 큐를 활용해 쉽게 해당 문제를 풀 수 있게 됩니다.

 

위에 해석한 개념을 코드로 반영하면 이렇게 구현할 수 있습니다. 먼저 Python에서 큐를 사용하기 위해 deque 라이브러리를 사용합니다. 

더불어 입력을 받아 var 변수에 저장하고, 리스트 형태의 deque를 만듭니다.

from collections import deque

var = int(input())
queue = deque()

그다음 리스트 형태의 큐에 입력한 정수값만큼 순차적으로 넣어줍니다.

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

queue가 하나가 남을 때까지 반복할 수 있도록 while문을 작성합니다. 또한 조건으로 홀수 짝수를 구분하여 카드를 버리거나 밑으로 보내는 로직을 구현합니다. 

while True:
    if len(queue) == 1:
        answer = queue.popleft()
        print(answer)
        break
    else:
        if cnt % 2 == 0:
            next = queue.popleft()
            queue.append(next)
            cnt = cnt + 1
        else:
            dele = queue.popleft()

 

💡 최종 코드

from collections import deque

var = int(input())
queue = deque()

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

cnt = 1

while True:
    if len(queue) == 1:
        answer = queue.popleft()
        print(answer)
        break
    else:
        if cnt % 2 == 0:
            next = queue.popleft()
            queue.append(next)
            cnt = cnt + 1
        else:
            dele = queue.popleft()
            cnt = cnt + 1