백준 - 10828번(스택)

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

10828번: 스택

import sys #sys 모듈 호출

n = int(sys.stdin.readline())
arr = []

for i in range(n):
    x = sys.stdin.readline().split() #띄어쓰기를 기준으로 입력받음(리스트 형태로 자동 저장)

    if x[0] == 'push': #x[0]는 첫번째 str
        arr.append(x[1]) #x[1]는 두번째 str 즉 push일때 입력받은 숫자

    elif x[0] == 'pop':
        if len(arr) != 0:
            print(arr.pop())
        else:
            print(-1)

    elif x[0] == 'size':
        print(len(arr))

    elif x[0] == 'empty':
        if len(arr) == 0:
            print(1)
        else:
            print(0)

    elif x[0] == 'top':
        if len(arr) == 0:
            print(-1)
        else:
            print(arr[len(arr)-1])

 

띄어쓰기를 기준으로 리스트 형태로 받아들이면 편하다. 왜냐하면 x [0] == 'push'일 때만 x [1]이 존재하기 때문이다.

x [0]은 정상적인 입력에서 항상 명령어가 될 것이다.

배열의 크기가 0 즉 배열에 아무것도 없을 때 명령어가 실행될 때 조건에 맞게 출력을 잘 설정해주면 된다.

 파이썬 기준으로 input() 함수를 사용하면 시간 초과가 난다.

코딩 테스트를 보거나 할 때 print는 몰라도 입력 함수는 반드시 sys 모듈의 readline() 함수를 이용하는 습관을 들이도록 하자.