스택 2 C#

📝 문제

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

 

문제 설명

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

명령은 총 다섯 가지이다.

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

---

입출력 예

예제 입력 1

9
4
1 3
1 5
3
2
5
2
2
5

예제 출력 1

1
2
5
3
3
-1
-1

---

입력 예 설명

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

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

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

---

풀이 코드

static void Main(string[] args)
{
    int index = int.Parse(Console.ReadLine());

    Stack<string> stack = new Stack<string>();
    StringBuilder sb = new StringBuilder();

    for (int i = 0; i < index; i++)
    {
        string[] temp = Console.ReadLine().Split();

        switch (temp[0])
        {
            case "1":
                stack.Push(temp[1]);
                break;

            case "2":
                if (stack.Count > 0)
                    sb.Append(stack.Pop() + "\\n");
                else
                    sb.Append("-1" + "\\n");
                break;

            case "3":
                sb.Append(stack.Count + "\\n");
                break;

            case "4":
                if (stack.Count == 0)
                    sb.Append("1" + "\\n");
                else
                    sb.Append("0" + "\\n");
                break;

            case "5":
                if (stack.Count > 0)
                    sb.Append(stack.Peek() + "\\n");
                else
                    sb.Append("-1" + "\\n");
                break;

        }
    }
    Console.WriteLine(sb);
}

• 이 문제를 해석해보면, 입력한 값을 바탕으로 Stack의 함수를 이용해 풀어가면 된다.

✍️ 중요 포인트

• 그때그때 Console.WriteLine을 사용하면 시간 초과가 되었다. 그래서 StringBuilder를 사용해 속도를 단축하여 통과해야 한다.
• 1~5의 값을 받아 Stack의 함수를 사용할 줄 아는지 확인하는 간단한 문제이다.