[백준]10828번 스택 - 스택

 

import sys
a = int(sys.stdin.readline())
stk = []
for i in range(a) :
    command = sys.stdin.readline().strip()
    if(command.split()[0] == 'push'):
        stk.append(int(command.split()[1]))
    elif(command == 'pop'):
        if len(stk) > 0:
            print(stk.pop())
        else :
            print(-1)
    elif(command == 'size'):
        print(len(stk))
    elif(command == 'empty'):
        if len(stk) == 0 :
            print(1)
        else :
            print(0)
    elif(command == 'top'):
        if len(stk) > 0 :
            print(stk[-1])
        else :
            print(-1)

 

코드 설명

먼저, 명령의 개수 만큼 반복문을 수행하기 위한 a 라는 변수에 값을 입력 받는다. 그 후 빈 리스트를 생성한다.

반복문을 진행하면서, 명령어를 입력받고 명령어에 따라 구분지어 스택에서의 연산을 수행한다.

 

Point: 인수가 존재하는 명령어가 있기 때문에 command.split()[0]를 통하여 명령어를 인식하고 [1]을 이용하여 인수를 인식할 수 있다.

또한, 스택의 가장 위에 있는 수를 출력하기 위해 [-1] 인덱스를 활용하였다.