[Seop's의 코드풀이] Back_Joon 10828 스택 python

안녕하세요 섭섭군 입니다.

 

이번에는 백준 10828 문제를 가져왔는데요. 아주 기본적인 stack 문제이며 문제 제목도 "스택"이네요.

 

스택에 대한 이해가 있다면 가볍게 풀법한 문제입니다. 

 

문제를 보면 다음과 같습니다.

 

 

정말 스택에 관한 문제입니다. 저는 이 문제를 pyhon으로 풀어보았는데 저의 코드는 다음과 같습니다.

 

import sys
input = sys.stdin.readline


def soluction(command) :
    if command[0] == "push" :
        stack.append(int(command[1]))
    elif command[0] == "pop" :
        if stack :
            print(stack.pop())
        else :
            print(-1)
    elif command[0] == "size" :
        print(len(stack))
    elif command[0] == "empty" :
        if stack :
            print(0)
        else :
            print(1)
    elif command[0] == "top" :
        if stack :
            print(stack[-1])
        else :
            print(-1)
        

N = int(input())
stack = []
for _ in range(N) :
    cmd = list(map(str, input().rstrip().split(" ")))
    soluction(cmd)

 

간단히 if 문을 주어서 풀 수 있는 문제였습니다.

 

그럼 전 다음 문제를 풀때 뵙도록 하겠습니다.