본문 바로가기

백준

[백준] 10828 - 스택

반응형

1. 문제

 

2. 코드

import sys
#read 20
l = []
num = int(input())
for i in range(num):
    a = sys.stdin.readline()
    
    #push
    if a[0] == 'p' and a[1] == 'u':
        i, x = a.split()
        l.append(int(x))
    #pop
    elif a[0] == 'p':
        if len(l) == 0:
            print(-1)
        else:
            print(l.pop())
    #size
    elif a[0] == 's':
        print(len(l))
    #empty
    elif a[0] == 'e':
        if len(l) == 0:
            print(1)
        else:
            print(0)
    #top
    else:
        if len(l) == 0:
            print(-1)
        else:
            print(l[len(l)-1])

 

3. 설명

파이썬의 경우 stack을 따로 제공하진 않지만, list를 사용하여 stack처럼 구현할 수 있다.

파이썬을 사용한다면 아주 간단한 문제!

 

728x90
반응형