[백준] 28278번: 스택 2 / 파이썬

문제 링크

---

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

 

풀이 코드

---

import sys
st = []

for _ in range(int(sys.stdin.readline())):
    a = sys.stdin.readline().split()
    
    if a[0] == '1':
        st.append(a[1])

    elif a[0] == '2':
        if st:
            print(st.pop())
        else:
            print(-1)

    elif a[0] == '3':
        print(len(st))

    elif a[0] == '4':
        if st:
            print(0)
        else:
            print(1)

    elif a[0] == '5':
        if st:
            print(st[-1])
        else:
            print(-1)

코드 해석

---

그냥 input을 사용하면 런타임 에러 (IndexError)가 발생한다.

sys.stdin.readline을 사용하여 실행 시간을 단축시킨다.

 

 

체감 난이도: ★☆☆☆☆