백준(Baekjoon) 10828번 스택 문제 풀이 | Python 파이썬 알고리즘

스택

출처: www.acmicpc.net/problem/10828

10828번: 스택

문제

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

명령은 총 다섯 가지이다.

▶ 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

풀이

파이썬은 따로 stack 구조를 제공하지 않아 list를 이용하여 stack을 표현할 수 있다.

먼저 stack 클래스를 선언하였고, 그 안에 문제에서 주어진 push, pop, top(peek), empty(isEmpty), size 기능을 각각 구현하였다.

#스택(LIFO)
class stack:
    #리스트를 이용하여 스택 생성
    def __init__ (self):
        self.items = [] 
    def push (self, item):
        self.items.append(item)    
    def pop(self):
        #if Stack is not empty
        if self.isEmpty()==0:
            #pop and return 
            return self.items.pop(-1)
        else:
            return -1      
    def peek(self):             #스택에서 top의 값을 읽어온다
        if self.isEmpty()==0:
            return self.items[-1]
        else:
            return -1
    def isEmpty(self):          #스택이 비어있는지 확인
        if len(self.items)==0:
            return 1
        else:
            return 0
    def size(self):             #스택 크기 반환
        return len(self.items)
n=int(input())
stk,result=stack(),[]
for i in range(n):
    com=input()
    if 'push' in com:
        stk.push(int(com[5:]))
    elif com=='pop':
        result.append(stk.pop())
    elif com=='size':
        result.append(stk.size())
    elif com=='empty':
        result.append(stk.isEmpty())
    elif com=='top':
        result.append(stk.peek())
for i in result:
    print(i)

이렇게 작성한 뒤 백준에 제출하면 시간 초과가 뜬다.

따라서 사용 언어를 Python 3가 아닌 Pypy3로 바꾼 뒤 돌렸더니 

성공했다..!! pypy3를 지원하지 않는 경우도 간혹가다 있으니, 해결방법을 알아두면 좋을 것 같아 구글링으로 찾아보았다.

알아보니, input()으로 입력값을 받는 것보다 sys 모듈을 추가해서 입력을 sys.stdin으로 대체하면 시간 초과 문제가 해결되는 것 같다.