백준 문제풀이/자료구조

[백준] 10828 스택 -python

peach_h 2023. 1. 27. 13:07

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

 

10828번: 스택

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지

www.acmicpc.net

input() 썻다가 시간초과에 걸렸던 것 같다 ( 어제푼건데 그새 까먹은 )

이미 큐를 겪어서 그런가 금방 풀었다!

import sys 
num = int(input())
temp = []

for i in range(num):
    com = sys.stdin.readline().split()
    if com[0] == 'push':
        temp.append(com[-1])
    elif com[0] == 'pop':
        if len(temp) == 0:
            print('-1')
        else :
            print(temp.pop())
    elif com[0] == 'size':
        print(len(temp))
    elif com[0] == 'empty':
        if len(temp) == 0:
            print(1)
        else :
            print(0)
    else :
        if len(temp) == 0:
            print(-1)
        else :
            print(temp[-1])

'백준 문제풀이 > 자료구조' 카테고리의 다른 글

[백준] 1874 스택수열 - python  (0) 2023.01.27
[백준] 9012 괄호 -python  (0) 2023.01.27
[백준] 2164 카드2 -python  (0) 2023.01.27
[백준] 1158 요세푸스 문제 - python  (0) 2023.01.25
[백준] 18258 큐2 - python  (0) 2023.01.25