[C#] 백준 18258 큐2

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

 

문제

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

명령은 총 여섯 가지이다.

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

입력

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

출력

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

예제 입력 1

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

예제 출력 1

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

 

Queue

Queue는 선입선출(FIFO First In First Out) 구조를 가지는 데이터 구조이다. 이는 쌓이는 스택과는 다르게 데이터가 입력된 순서대로 처리되는 구조를 의미한다.

 

Queue의 함수에는 아래의 종류가 있다.

 

• Enqueue(T item): 큐에 요소를 추가. 이 때 추가되는 요소는 큐의 맨 뒤에 위치함.
• Dequeue(): 큐의 맨 앞에 있는 요소를 제거하고 반환.
• Peek(): 큐의 맨 앞에 있는 요소를 반환하지만 제거하지는 않는다..
• Count: 큐에 저장된 요소의 개수를 반환한다.
• Clear(): 큐의 모든 요소를 제거.
• Contains(T item): 큐에 특정 요소가 포함되어 있는지 여부를 확인.
• ToArray(): 큐의 모든 요소를 배열로 복사.
• TrimExcess(): 큐의 여유 공간을 제거하여 메모리를 최적화.

 

 

문제풀이

 

저번 스택과 마찬가지로 if문을 통해 어떤 문자열이 들어왔는지에 따라 queue의 값을 가져오면된다.

앞에서 했던 스택 내용과 비슷하기 때문에 상세한 풀이는 생략하겠다.

 

시간초과

using System.Text;

namespace CodeTest
{
    internal class Program
    {
        static void Main(string[] args)
        {
            int count = int.Parse(Console.ReadLine());

            Queue<int> queue = new Queue<int>();
            StringBuilder cw = new StringBuilder();

            for(int i = 0; i < count; i++) 
            {
                string[] input= Console.ReadLine().Split(" ");
            
                // push
                if(input.Contains("push"))
                {
                 	lastPush = int.Parse(input[1]);
                    queue.Enqueue(lastPush);
                }
                else if(input.Contains("pop"))
                {
                    if(queue.Count == 0)
                        cw.AppendLine("-1");
                    else
                        cw.Append(queue.Dequeue() + "\n");
                }
                else if(input.Contains("size"))
                {
                    cw.Append(queue.Count() + "\n");
                }
                else if(input.Contains("empty"))
                {
                    if (queue.Count == 0)
                        cw.AppendLine("1");
                    else
                        cw.AppendLine("0");

                }
                else if(input.Contains("front"))
                {
                    if (queue.Count == 0)
                        cw.AppendLine("-1");
                    else
                    {
                        cw.Append(queue.Peek() + "\n");
                    }
                }
                else if (input.Contains("back"))
                {
                    if (queue.Count == 0)
                        cw.AppendLine("-1");
                    else
                    {
                        cw.Append(queue.Last() + "\n");
                    }
                }

            }

            Console.WriteLine(cw);
        }
    }
}

 

 

 

원인

원인은 queue.Last()에 있었다. 이는 반복될 경우 비용이 비싼 데 이유는 queue의 마지막 부분을 가져오려고 하기 때문에 계속 반복될 경우 O(n)의 시간 복잡도가 반복되기 때문에 시간초과가 일어납니다.

 

정답

using System.Text;

namespace CodeTest
{
    internal class Program
    {
        static void Main(string[] args)
        {
            int lastPush = 0;
            int count = int.Parse(Console.ReadLine());

            Queue<int> queue = new Queue<int>();
            StringBuilder cw = new StringBuilder();

            for(int i = 0; i < count; i++) 
            {
                string[] input= Console.ReadLine().Split(" ");
            
                // push
                if(input.Contains("push"))
                {
                    lastPush = int.Parse(input[1]);
                    queue.Enqueue(lastPush);
                }
                else if(input.Contains("pop"))
                {
                    if(queue.Count == 0)
                        cw.AppendLine("-1");
                    else
                        cw.Append(queue.Dequeue() + "\n");
                }
                else if(input.Contains("size"))
                {
                    cw.Append(queue.Count() + "\n");
                }
                else if(input.Contains("empty"))
                {
                    if (queue.Count == 0)
                        cw.AppendLine("1");
                    else
                        cw.AppendLine("0");

                }
                else if(input.Contains("front"))
                {
                    if (queue.Count == 0)
                        cw.AppendLine("-1");
                    else
                    {
                        cw.Append(queue.Peek() + "\n");
                    }
                }
                else if (input.Contains("back"))
                {
                    if (queue.Count == 0)
                        cw.AppendLine("-1");
                    else
                    {
                        cw.Append(lastPush + "\n");
                    }
                }

            }

            Console.WriteLine(cw.ToString());
        }
    }
}

 

해결방법은 Last를 사용하지 않고 이를 push할 때 값을 저장해서 추후에 출력하는 방식을 했더니 해결되었다.

 

 

회고

queue에 대한 내용을 더 잘 알 수 있었으며, Last의 시간복잡도가 반복될 경우 비용에 문제가 생긴다는 것에 대해 알게되었습니다.