C++ BAEKJOON 28278번 : 스택2

[C++] BAEKJOON 28278번 : 스택2

 

시간                 제한메모리 제한                       제출                     정답                맞힌 사람               정답 비율

 

2 초

1024 MB

6774

2693

2259

40.622%

문제

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 다섯 가지이다.

1. 1 X: 정수 X를 스택에 넣는다. (1 ≤ X ≤ 100,000)
2. 2: 스택에 정수가 있다면 맨 위의 정수를 빼고 출력한다. 없다면 -1을 대신 출력한다.
3. 3: 스택에 들어있는 정수의 개수를 출력한다.
4. 4: 스택이 비어있으면 1, 아니면 0을 출력한다.
5. 5: 스택에 정수가 있다면 맨 위의 정수를 출력한다. 없다면 -1을 대신 출력한다.

입력

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

둘째 줄부터 N개 줄에 명령이 하나씩 주어진다.

출력을 요구하는 명령은 하나 이상 주어진다.

 

출력

출력을 요구하는 명령이 주어질 때마다 명령의 결과를 한 줄에 하나씩 출력한다.

 

문제풀이

명령의 이해

명령 1 -  정수 X를 스택에 넣는다. (1 ≤ X ≤ 100,000)

            >> 명령으로 1이 들어왔을 경우 입력을 하나 더 받아 넣는다.

명령 2 -  스택에 정수가 있다면 맨 위의 정수를 빼고 출력한다. 없다면 -1을 대신 출력한다.

            >> 명령으로 2가 들어왔을 경우 스택이 비어있는지 확인, top()을 출력하고 pop() 빼주기

명령 3 -  스택에 들어있는 정수의 개수를 출력한다.

            >> 명령으로 3이 들어왔을 경우 size()를 출력한다.

명령 4 -  스택이 비어있으면 1, 아니면 0을 출력한다.

            >> 명령으로 4가 들어오면 empty()를 확인하고 1또는 0을 출력한다.

명령 5 -  스택에 정수가 있다면 맨 위의 정수를 출력한다. 없다면 -1을 대신 출력한다.

            >> 명령으로 5가 들어오면 empty()를 확인하고 top()을 출력한다.

 

시간 초과

문제 : cin, cout, endl 명령들이 시간을 많이 잡아먹는다. (입출력 버퍼를 지우면서 작동)

 

해결 1 : C++과 C의 표준 스트림의 동기화를 끊어주는 ios_base::syn_with_stdio(false) 또는 ios_base::syn_with_stdio(0)을 활용하여 시간을 단축해준다.

해결 2 : cin과 cout의 결합을 끊어주기 위해  cin.tie(NULL) 또는 cin.tie(0)을 활용

           >> 입출력이 반복될 때 버퍼를 계속 비우느라 시간 잡아 먹는다.

해결 3 : endl 대신 "\n" 사용, endl명령은 개행과 더불어 출력 버퍼를 비우는 flush 역할도 하기 때문

 

 

#include <iostream>
#include <stack>

using namespace std;

void printCommand(int result)
{
	cout << result << "\n";
}

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

	stack<int> s;
	int n;

	cin >> n;

	for (int i = 0; i < n; i++)
	{
		int cmd;
		cin >> cmd;
		
		switch (cmd)
		{
			case 1 :
				int x;
				cin >> x;
				s.push(x);
				break;
			case 2 :
				if (s.empty())
					printCommand(-1);
				else
				{
					printCommand(s.top());
					s.pop();
				}
				break;
			case 3 :
				printCommand(s.size());
				break;
			case 4:
				printCommand(s.empty() ? 1 : 0);
				break;
			case 5:
				printCommand(s.empty() ? -1 : s.top());
				break;
		}
	}
}