[Python][백준 10828] 스택

📚 스택 문제 풀이

백준 10828번 스택 문제의 파이썬 풀이

📝 문제 설명

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

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

💡 풀이

이 문제는 스택의 기본적인 연산들을 구현하는 문제이다. 파이썬의 리스트를 사용하여 스택을 구현하고, 각각의 명령어에 따라 적절한 동작을 수행하면 된다.

🔍 코드 설명

1. 명령의 수 N을 입력받는다.
2. N번 반복하면서:
   • 명령어를 입력받아 분석한다.
   • 각 명령어에 맞는 동작을 수행한다.
   • 결과를 출력해야 하는 명령의 경우 결과값을 저장한다.
3. 저장된 모든 결과를 한 번에 출력한다.

✨ 참고

• 파이썬의 리스트는 스택의 모든 연산을 지원한다.
• append()로 요소를 추가하고 pop()으로 제거한다.
• sys.stdin.readline()을 사용하여 입력 속도를 개선할 수 있다.
• sys.stdout.write()를 사용하여 출력 속도를 개선할 수 있다.

🎯 주의사항

• 스택이 비어있을 때 pop이나 top 명령이 들어올 수 있다.
• 시간 제한이 있으므로 입출력 최적화가 필요하다.

---

📝 코드

import sys

n = int(sys.stdin.readline().strip())
stack = []
results = []

for _ in range(n):
    command = sys.stdin.readline().strip().split()
    
    if command[0] == "push":
        stack.append(int(command[1]))
    elif command[0] == "pop":
        results.append(stack.pop() if stack else -1)
    elif command[0] == "size":
        results.append(len(stack))
    elif command[0] == "empty":
        results.append(0 if stack else 1)
    elif command[0] == "top":
        results.append(stack[-1] if stack else -1)

sys.stdout.write("\n".join(map(str, results)) + "\n")

Python Coding Test 카테고리 내 다른 글 보러가기