백준 10828번 풀이

스택 실습 예제입니다. (실버 IV)

문제 내용

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

명령은 총 다섯 가지이다.

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

조건

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)
둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. (1 ≤ 정수 ≤ 100,000)

시간제한: 0.5초
메모리제한: 256 MB

풀이 과정

문제 이해

해당 문제는 이해 단계가 필요 없이 단순히 스택처럼 구현하는 것이었습니다.

과정 풀이

1. N 입력 후 N개 만큼 명령어commands 받기
2. stack 생성하기
3. commands로 반복을 돌며, 각 경우의 따른 처리 하기
   3-1. push : stack.append(숫자)
   3-2. top : stack[-1]로 판단하기
   3-3. size : python에서 제공하는 len(stack) 활용
   3-4. pop : stack.pop() 활용
   3-5. empty : len(stack) 활용

위 과정으로 진행을 하였습니다.

정답

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

commands = []
stack = []

for i in range(0, N):
    commands.append(input().split())

for command in commands:
    if command[0] == "push":
        stack.append(command[1])
        
    elif command[0] == "top":
        if len(stack) <= 0:
            print(-1)
            continue
        print(stack[-1])
        
    elif command[0] == "size":
        print(len(stack))
        
    elif command[0] == "pop":
        if len(stack) <= 0:
            print(-1)
            continue
        print(stack.pop())
        
    elif command[0] == "empty":
        if len(stack) > 0:
            print(0)
        else:
            print(1)

위 코드를 통해 성공했다.
(시도: 1, 정답: 1)

달성 조건

Language	Memory	Time
Python3	34324 KB	52 ms