[백준][파이썬] 2161번 카드1

 2161번 : 카드1

문제

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

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

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

N이 주어졌을 때, 버린 카드들을 순서대로 출력하고, 마지막에 남게 되는 카드를 출력하는 프로그램을 작성하시오.

입력

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

출력

첫째 줄에 버리는 카드들을 순서대로 출력한다. 제일 마지막에는 남게 되는 카드의 번호를 출력한다.

링크

2161번: 카드1 (acmicpc.net)

2161번: 카드1

 

 접근 방법

1. 카드의 수(N) 입력 받고
2. 카드를 리스트로 생성
3. 카드의 수가 0이 될 때까지 (while문)
4. 카드를 버리고 옮기는 규칙 실행
   
   1. 버리는 카드는 pop으로 원소 삭제
   2. 옮기는 카드는 pop하여 append로 원소 이동

+ 큐 참고 자료 : [자료구조] 큐, 스택 / queue, stack — y-seo의 딩코 기록들 (tistory.com)

[자료구조] 큐, 스택 / queue, stack

 

 풀이

 

1. 리스트 안 for문 (List Comprehension)

입력 받은 카드의 수(n)만큼 리스트 원소를 1부터 n까지 추가해주어야 했는데 처음에는 다음과 같이 코드를 작성하였다.

card = [] #리스트 생성
for i in range(1,n+1):
    card.append(i) #리스트에 원소 추가

그러나 이 3줄을 리스트 안 for문 사용으로 1줄로 단축하여 코드를 작성할 수 있다.

변수명 = [ 식 for 변수 in 리스트 ]

다음 코드를 통해 리스트에서 생성한 숫자를 변수에 꺼내고 식을 이용해 리스트를 생성한다.

따라서 이 문제에서는 다음과 같이 사용하였다.

card = [ i for i in range(1, n+1)]

range(1, n+1)에서 생성된 숫자 1~n을 변수 i에 넣고 최종적으로 i를 이용해 리스트를 생성하는 방식이다.

 

 전체코드

import sys

n = int(sys.stdin.readline()) #카드 수 입력받기
card = [ i for i in range(1, n+1)] #카드 생성
while len(card) != 1: 
    print(card.pop(0), end=' ') #버리는 카드 출력
    card.append(card.pop(0)) #카드 옮기기
print(card[0]) #마지막 카드 출력