[백준/ Python] 28278. 스택2

문제

 

풀이 코드 

기본적으로 스택 연산 사용할 줄 알면, 쉽게 풀리는 문제였다!
대신, 입력 양이 너무 커지면 문제가 생기길래 이 부분에 대해 알아보니, 

import sys 
n = int(sys.stdin.readline()) 

이걸 사용하면 빠르게 입력을 받을 수 있다고 한다.
풀다가 영 답이 안 보이면 이렇게도 해봐야지 

#N = int(input())
import sys
N = int(sys.stdin.readline())
stack = []
for i in range(N):
    n = sys.stdin.readline().split()
    # n, x = map(int, input().split(" "))
    if n[0] == "1":
        stack.append(n[-1])
    elif n[0] == "2":
        if stack: print(stack.pop())
        else: print("-1")
    elif n[0] == "3":
        print(len(stack))
    elif n[0] == "4":
        if stack: print("0")
        else: print("1")
    elif n[0] == "5":
        if stack: print(stack[-1])
        else: print("-1")