몽상실현개발주의

[BOJ] 10828 / 스택 / Python 파이썬 본문

Algorithm PS/BOJ

[BOJ] 10828 / 스택 / Python 파이썬

migrationArc 2021. 5. 23. 15:42

[BOJ] 10828 / 스택 / Python 파이썬

[BOJ] 10828 / 스택 / Python 파이썬

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

 

10828번: 스택

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지

www.acmicpc.net

 

풀이

Stack 을 구현하는 문제이다.

Python 의 List 를 활용하면 된다.

 

import sys

N = int(sys.stdin.readline())

stack = []

for _ in range(N):
    order = sys.stdin.readline().split()
    if order[0] == 'push':
        stack.append(int(order[1]))
    elif order[0] == 'pop':
        if stack:
            print(stack.pop())
        else:
            print(-1)
    elif order[0] == 'size':
        print(len(stack))
    elif order[0] == 'empty':
        if stack:
            print(0)
        else:
            print(1)
    elif order[0] == 'top':
        if stack:
            print(stack[-1])
        else:
            print(-1)

 

Comments