[BOJ] 10828번 : 스택

■ 문제풀이

· push -> append

· pop -> pop

· size -> len

· top -> stack[-1]

 

'if not stack:'

stack이 비어있을 경우 False를 반환함으로 if not 을 사용하여 print(1) 수행

 

■ 문제해답

import sys
input = sys.stdin.readline

n = int(input())

stack = []
for _ in range(n):
    data = input().split()
    
    if data[0] == 'push':
        stack.append(data[1])
    elif data[0] == 'pop':
        if not stack:
            print(-1)
        else:
            print(stack.pop())
    elif data[0] == 'size':
        print(len(stack))
    elif data[0] == 'empty':
        if not stack:
            print(1)
        else:
            print(0)
    elif data[0] == 'top':
        if not stack:
            print(-1)
        else:
            print(stack[-1])

 

문제출처 : https://www.acmicpc.net/problem/10828

10828번: 스택