백준 10828번 : 스택(S4)

 

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

10828번: 스택

 

 

✨ 문제

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

명령은 총 다섯 가지이다.

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

 

✨ 접근 방법

 

- 문제에 적혀있듯이, 스택을 구현하는 문제이다.

 

- 파이썬의 list를 이용해서 stack을 정의한다.

 

- push는 파이썬의 append 함수, pop은 pop 함수, size는 len 함수, empty는 `if stack`, top은 index -1를 통해 접근해서 출력한다.

 

- 단, 스택에 정수가 없을 경우를 확인하는 예외 처리를 해줘야 한다. 

 

 

 

✨ 코드

import sys

def input():
    return sys.stdin.readline().rstrip()

def main():
    N = int(input())
    print(N)
    stack = []

    for _ in range(N):
        command = input().split()
        num = 0
        if len(command) == 2:
            num = command[1]
        command = command[0]

        len_stack = len(stack)

        if command == 'push':
            stack.append(num)
        elif command == 'pop':
            if len_stack == 0:
                print(-1)
            else :
                print(stack.pop())
        elif command =='size':
            print(len_stack)
        elif command == 'empty':
            if len_stack == 0:
                print(1)
            else :
                print(0)
        elif command == 'top' :
            if len_stack == 0:
                print(-1)
            else :
                print(stack[len_stack-1])