백준 - 10828- 스택 - Python/스택

10828번: 스택 (acmicpc.net)

10828번: 스택

문제

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

명령은 총 다섯 가지이다.

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

입력

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

출력

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

 

 

풀이

문제랑 푸는 방법은 생각했는데 명령어를 어떻게 입력해야할지 몰랐다.

아직까지 은근히 입력에 대한 어려움이 있다. 씨플플같은 경우에는 문자열 입력하는게 좀 까다롭?기 때문에 파이썬도 그러려나 했는데 은근 쉬운것 같기도 하고.. ㅎ 

import sys
input = sys.stdin.readline
n = int(input())
stk = [] 
for _ in range(n):
    x = list(map(str,input().split()))
    if x[0] =='push':
        stk.append(int(x[1]))
    elif x[0] =='pop':
        if stk:
            print(stk.pop())
        else:
            print('-1')
    elif x[0] =='size':
        print(len(stk))
    elif x[0] =='empty':
        if stk:
            print('0')
        else:
            print('1')
    elif x[0] =='top':
        if stk:
            print(stk[-1])
        else:
            print('-1')

input대신에 값을 입력받는 데 더 빠른 readline을 쓰면 된다.

백준 문제에서 시간초과가 날 때는 대부분 readline을 쓰거나 PyPy로 제출하면 해결된다.

 

 

사용된 개념

 

import sys

input = sys.stdin.readline을 입력해두면 input 함수를 사용하는 것처럼 입력하면서 readline을 이용할 수 있다.