[백준 10828] 스택

Junyoung Park·2022년 2월 25일
0

코딩테스트

목록 보기
80/631
post-thumbnail

1. 문제 설명

스택

2. 문제 분석

스택 기능을 구현하는 문제

  • input()이 아니라 sys.stdin.readline()을 사용해야 시간 초과가 나지 않는다.

3. 나의 풀이

import sys

n = int(input())
stack = []
for _ in range(n):
    cmd = sys.stdin.readline().strip()

    if cmd == 'size':
        print(len(stack))
    elif cmd == 'empty':
        if stack: print(0)
        else: print(1)
    elif cmd == 'top':
        if stack: print(stack[-1])
        else: print(-1)
    elif cmd == 'pop':
        if stack: print(stack.pop(-1))
        else: print(-1)
    else:
        cmd, num = cmd.split()
        num = int(num)
        stack.append(num)
profile
JUST DO IT

0개의 댓글