본문 바로가기

PS_Baekjoon

[백준 C++] 2161번 : 카드1

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

 

2161번: 카드1

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

www.acmicpc.net

풀이 :

- n이 주어지고 1부터 n까지 배열에 넣는다

- 문제에서 요구하는 대로 제일 앞에 있는 카드를 버리고, 그다음 앞에 있는 카드를 뒤로 옮겨준다. 

- 버려진 카드를 출력한 뒤 마지막 남은 카드를 출력하라고 하는데,

   따로 처리해 줄 필요 없이 마지막 카드도 버리고 같이 한번에 출력해주면 된다.

- 스택의 구현과 다름없는 문제로, queue를 이용하면 쉽게 풀 수 있다.

 

코드 :

#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std;

int main(){
    cin.tie(NULL);
    ios::sync_with_stdio(false);

	int n;
	queue<int> cards;
	vector<int> trash;

	cin >> n;
	for(int i=1; i<=n; i++){
		cards.push(i); //1부터 n까지 카드를 넣어준다
	}
	while(!cards.empty()){ //카드가 다 없어질 때까지 while문을 돌려준다
		trash.push_back(cards.front()); //제일 위(앞)에 있는 카드를 버리는 배열로 옮긴다
		cards.pop(); //옮긴 후 배열에서 삭제
		if(cards.empty()){ //만약 배열이 비었다면 break
			break;
		}else{
			cards.push(cards.front()); //제일 위(앞)에 있는 카드를 뒤로 옮겨준다
			cards.pop(); 
		}
	}
	for(int i=0; i<trash.size(); i++){
		cout << trash[i] << ' '; 
	}
	cout << '\n';
}