[백준 10828번] 스택 파이썬 풀이

스택

0.5 초 (추가 시간 없음)

256 MB

245023

88329

64207

37.338%

문제

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 다섯 가지이다.

• push X: 정수 X를 스택에 넣는 연산이다.
• pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 스택에 들어있는 정수의 개수를 출력한다.
• empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
• top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

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

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

예제 입력 1

14
push 1
push 2
top
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
top

예제 출력 1

2
2
0
2
1
-1
0
1
-1
0
3

예제 입력 2

7
pop
top
push 123
top
pop
top
pop

예제 출력 2

-1
-1
123
123
-1
-1

 

스택에 대해 알고있는지, 자신의 언어로 스택을 구현하여 사용할 수 있는지를 묻는 문제이다.

새로운 언어를 연습할때 풀기에 좋은 문제라고 생각된다.

 

스택이 무엇인지 잘 모른다면 아래 게시글을 참고하길 바란다.

 

https://76codefactory.tistory.com/28

 

스택에 대해 알아보자

 

 

파이썬은 리스트가 배열과 스택의 역할을 동시에 하므로, 문제의 조건에 맞춰서 코드를 작성하면 되는 아주 간단한 문제이다.

주의해야 할점은 명령을 입력받는데 걸리는 시간을 단축하지 않으면 시간초과 판정을 받을 수 있으므로 수많은 데이터를 입력받을때는

sys의 stdin.readline을 통해 입력받아야 시간을 단축시킬 수 있다. 또한 언어를 선택할시 python3가 아닌 pypy3를 선택해야지 정답 판정을 받을 수 있다.

 

스택과 파이썬의 리스트에 대해 알고 있다면, 별도의 해설이 필요하지 않다고 판단하여 코드의 주석으로 설명을 대체하겠다.

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

# 시간초과판정을 받지 않기 위하여(입력에 걸리는 시간을 단축하기 위해)
import sys
input = sys.stdin.readline

# 스택의 역할을 할 리스트 생성
stack = []

# 명령의 횟수
n = int(input())

# len 명령을 최대한 줄이기 위해 스택의 크기를 측정할 변수 생성
count = 0

for i in range(n):
    # 명령 입력받기
    command = list(map(str, input().split()))
    # push 명령이 존재한다면 숫자와 분할
    if command[0] == "push":
        stack.append(int(command[1]))
        count += 1  # 스택 내용물 추가
    elif command[0] == "top":
        # 스택이 비어있다면 -1 출력
        if count == 0:
            print(-1)
        else:
            # 가장 위에 있는 정수 출력
            print(stack[-1])
    elif command[0] == "size":
        # 스택의 크기를 출력한다.
        print(count)
    elif command[0] == "empty":
        # 스택이 비어있으면 1을 출력하고 아니면 0을 출력한다.
        if count == 0:
            print(1)
        else:
            print(0)
    elif command[0] == "pop":
        # 스택이 비어있다면
        if count == 0:
            print(-1)
        else:  # 스택이 비어있지 않다면
            print(stack.pop())
            count -= 1

 

 

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