[BOJ/Python, Java] 백준 2164번 - 카드2

백준 #2164 카드2

2164번: 카드2

 

Python ver.

from collections import deque

n = int(input())
queue = deque([i+1 for i in range(n)])

ch = 1  # 어떤 동작 해야 하는지 표시 (1: 제일 위에 있는 카드 제거, 2: 제일 위에 있는 카드 맨 밑으로)
cnt = n
while cnt != 1:
    if ch == 1:
        queue.popleft()
        ch = 2
        cnt -= 1
    elif ch == 2:
        queue.append(queue.popleft())
        ch = 1

print(queue[0])

 

Java ver.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayDeque;

public class Main {
	
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		int n = Integer.parseInt(br.readLine());
		
		ArrayDeque<Integer> q = new ArrayDeque<Integer>();
		for (int i = 0; i < n; i++) {				
			q.addLast(i+1);
		}
		
		/** 어떤 동작 해야 하는지 표시 
		 * 1: 제일 위에 있는 카드 제거, 2: 제일 위에 있는 카드 맨 밑으로 */
		int ch = 1;
		int cnt = n;
		
		while (cnt != 1) {
			if (ch == 1) {
				q.pollFirst();
				ch = 2;
				cnt--;
			} else {
				q.addLast(q.pollFirst());
				ch = 1;
			}
		}
		
		System.out.println(q.pollFirst());
		
	}
		
}

자료구조만 이용하면 풀 수 있는 문제!