[Algorithm] [Python] BOJ/백준 - 10828_스택

[10828] 스택

---

문제

---

시간 제한 메모리 제한 제출 정답 맞은 사람 정답 비율
0.5 초 (추가 시간 없음) 256 MB 65199 25059 18345 40.037%
문제
정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 다섯 가지이다.

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

입력

---

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

출력

---

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

문제 풀이

---

우리가 아는 자료구조 스택의 주요 기능들을 구현한다.

지원해야 하는 기능은 총 5가지 push, pop, size, empty, top이다.

1차 제출 코드

---

def push(elem):
    stack.append(elem)

def empty():
    result = 1 if size() == 0 else 0
    return result

def pop():
    try:
        print(stack.pop())
    except:
        print("-1")

def size():
    return len(stack)

def top():
    if empty() == 1:
        print("-1")
    else:
        print(stack[len(stack)-1])

stack = []
n = int(input())

for _ in range(n):
    cmd = input().split()

    if cmd[0] == 'push':
        push(cmd[1])

    elif cmd[0] == 'pop':
        pop()

    elif cmd[0] == 'empty':
        print(empty())

    elif cmd[0] == 'size':
        size()

    elif cmd[0] == 'top':
        top()

첫번쨰로 제출한 코드는 시간 초과가 떳다..

2차 제출 코드

---

import sys

def push(elem):
    stack.append(elem)

def empty():
    result = 1 if size() == 0 else 0
    return result

def pop():
    try:
        print(stack.pop())
    except:
        print("-1")

def size():
    return len(stack)

def top():
    try:
        print(stack[-1])
    except:
        print("-1")

stack = []
n = int(sys.stdin.readline())

for _ in range(n):
    cmd = sys.stdin.readline().split()
    #cmd = input().split()

    if cmd[0] == 'push':
        push(cmd[1])

    elif cmd[0] == 'pop':
        pop()

    elif cmd[0] == 'empty':
        print(empty())

    elif cmd[0] == 'size':
        print(size())

    elif cmd[0] == 'top':
        top()

2번째 작성한 코드에서는 sys 모듈을 추가하여 입력받는 부분을 sys.stdin으로 대체하였다.

각각의 기능들을 입력에 따라 진행해야하니 바꿔줘도 바로 풀렸다.