[18528] C++ : 큐 2

이번에는 큐 기본문제를 풀어보았다.

문제

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

명령은 총 여섯 가지이다.

• push X: 정수 X를 큐에 넣는 연산이다.
• pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 큐에 들어있는 정수의 개수를 출력한다.
• empty: 큐가 비어있으면 1, 아니면 0을 출력한다.
• front: 큐의 가장 앞에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• back: 큐의 가장 뒤에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 2,000,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

예제 입력

15
push 1
push 2
front
back
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
front

예제 출력

1
2
2
0
1
2
-1
0
1
-1
0
3

 

풀이

스택이랑 문제가 유사해서 그냥 같은 형식으로 문제를 해결했다.

#include<iostream>
#include<queue>
#include<string>

using namespace std;

int main()
{
	int repeat;
	string command;
	queue<int> q;

	cin >> repeat;
	for (int i = 0; i < repeat; i++)
	{
		cin >> command;
		if (command.compare("push") == 0)
		{
			int number;
			cin >> number;
			q.push(number);
		}
		else if (command.compare("pop") == 0)
		{
			if (q.empty())
				cout << -1 << endl;
			else {
				cout << q.front() << endl;
				q.pop();
			}
		}
		else if (command.compare("size") == 0)
		{
			cout << q.size() << endl;
		}
		else if (command.compare("empty") == 0)
		{
			if (q.empty())
				cout << 1 << endl;
			else
				cout << 0 << endl;
		}
		else if(command.compare("front") == 0)
		{
			if (q.empty())
				cout << -1 << endl;
			else 
				cout << q.front() << endl;
		}
		else {
			if (q.empty())
				cout << -1 << endl;
			else
				cout << q.back() << endl;
		}
	}

	return 0;
}

 

이렇게 간단하게 끝이 나나 싶었는데..... 

시간초과가 발생했다. 시간초과가 발생할 수 있는 부분을 생각해보니 compare 부분이 의심되었다. 그래서 찾아보니 compare 대신 그냥 == 을 사용해도 비교가 된다는 사실을 알게 되어 그냥 비교로 바꿨다. 

하지만, 그래도 해결이 되지 않았다. 그 다음 의심이 되는 부분은 입출력 부분이다. 자바로 해결한 사람들의 포스팅을 읽어보니 대부분 입출력 문제였다. 그래서 검색해본 결과 아래 코드를 얻을 수 있었다. 

cin.tie(NULL);
cin.sync_with_stdio(false);

이렇게 하면 입출력 시간이 빨라진다고 한다. 그리고 endl 대신 "\n" 을 사용했다. 

 

최종 코드는 아래와 같다. 

#include<iostream>
#include<queue>
#include<string>

using namespace std;

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

	int repeat;
	string command;
	queue<int> q;
	int number;
	bool empty;

	cin >> repeat;
	for (int i = 0; i < repeat; i++)
	{
		empty = q.empty();
		cin >> command;
		if (command == "push")
		{
			cin >> number;
			q.push(number);
		}
		else if (command == "pop")
		{
			if (empty)
				cout << -1 << "\n";
			else {
				cout << q.front() << "\n";
				q.pop();
			}
		}
		else if (command == "size")
		{
			cout << q.size() << "\n";
		}
		else if(command == "empty")
		{
			if (empty) {
				cout << 1 << "\n";
			}
			else 
			{
				cout << 0 << "\n";
			}
		}
		else if (command == "front")
		{
			if (empty) 
				cout << -1 << "\n";
			else
				cout << q.front() << "\n";
		}
		else
		{
			if (empty)
				cout << -1 << "\n";
			else
				cout << q.back() << "\n";
		}

	}

	return 0;
}

 

이렇게 하니 통과할 수 있었다. 

 

백준이 조금 까다로운게 작동이 잘 되어도 시간 초과가 자주 된다는 것이다. 아직 효율적인 알고리즘을 짜는 것에 약한 나에겐 아직 배워갈 점이 많은 것 같다. 오늘도 이렇게 하나씩 풀어간다.