백준 #10828

7386 단어 백준스택백준

python

import sys
n = int(input())

# 스택 생성
stack = []

# 정수 x를 스택에 넣는다.
def s_push(x: int):
    stack.append(int(x))

# 스택의 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 스택이 비어있으면 -1을 출력한다.
def s_pop():
    if len(stack) == 0:
        return -1
    return stack.pop()

# 스택의 사이즈를 출력한다.
def s_size():
    return len(stack)

# 스택이 비어있으면 1, 아니면 0을 출력한다.
def s_empty():
    if len(stack) == 0:
        return 1
    else:
        return 0

# 스택의 가장 위에 있는 정수를 출력한다. 스택이 비어있으면 -1을 출력한다.
def s_top():
    if len(stack) == 0:
        return -1
    return stack[-1]

for i in range(n):
    command = sys.stdin.readline().split()  # input을 통해 입력받을 시 시간 초과가 된다.
    if command[0] == "push":
        s_push(command[1])
    elif command[0] == "pop":
        print(s_pop())
    elif command[0] == "size":
        print(s_size())
    elif command[0] == "empty":
        print(s_empty())
    elif command[0] == "top":
        print(s_top())

좋은 웹페이지 즐겨찾기