[BOJ/Python] 10828. 스택

📌 문제

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

10828번: 스택

 

📌 문제 접근 방법

1. 함수로 따로 만들어보기
2. 조건문으로 구현
3. → 조건문이 시간이 더 짧았다! 왜일까..?

 

📌 코드

import sys
    
N = int(sys.stdin.readline())

stack = []

for _ in range(N) :
    command = sys.stdin.readline()[:-1]

    if 'push' in command :
        number = int(command.split()[-1])
        stack.append(number)

    elif command == 'pop' :
        if stack :
            print(stack.pop())
        else :
            print(-1)

    elif command == 'size' :
        print(len(stack))

    elif command == 'empty' :
        if stack :
            print(0)
        else :
            print(1)

    elif command == 'top' :
        if stack :
            print(stack[-1])
        else :
            print(-1)