[백준] 10828번 파이썬(Python) feat.스택

문제

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

명령은 총 다섯 가지이다.

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

입력

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

출력

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

---

'정답 코드'

n = int(input())

orders = []
stack = []

for _ in range(n):
    orders.append(list(input().split()))

for i in range(n):
    if orders[i][0] == 'push':
        stack.append(orders[i][1])
    elif orders[i][0] == 'pop':
        if len(stack) == 0:
            print(-1)
        else:
            pop = stack.pop()
            print(pop)
    elif orders[i][0] == 'size':
        print(len(stack))
    elif orders[i][0] == 'empty':
        if len(stack) == 0:
            print(1)
        else:
            print(0)
    elif orders[i][0] == 'top':
        if len(stack) == 0:
            print(-1)
        else:
            top = stack.pop()
            stack.append(top)
            print(top)

 

'공부한 것'

- 스택(Stack) : 파이썬에서는 스택 자료구조를 따로 제공하지 않고, 기본 클래스인 List를 통해 구현함

~> 후입선출(LIFO) ex) 상자 쌓기

~> 주요 메소드

1. push : 데이터 삽입, append() 함수를 이용하여 매개변수를 리스트(스택) 맨 뒤에 넣음
2. pop : 가장 마지막에 삽입한 데이터를 삭제, pop() 함수를 이용하여 리스트의 맨 뒤 요소를 꺼내고 삭제
3. top : 가장 마지막에 삽입한 데이터를 삭제하지 않고 return 해주는 메소드, 리스트[-1]을 사용해 top 원소 가져옴
4. isEmpty : 현재 스택이 비어있는지 여부 확인, 'if not 리스트:', 'if 리스트:' 구문 사용해 empty 확인

stack = []

# push
stack.append(1)
print("push 후 스택 :", stack)

stack.append(2)
print("push 후 스택 :", stack)

# pop
pop = stack.pop()
print("pop한 요소 :", pop)
print("pop 후 스택 :", stack)

# top (1)
top = stack.pop()
stack.append(top)
print("top 요소: ", top)
print("top 확인 후 스택:", stack)

# top (2)
print("top 요소: ", stack[-1])
print("top 확인 후 스택:", stack)

if not stack:
    print("스택 비어있음")
else:
    print("스택 비어있지 않은 상태")
    print("현재 스택 :", stack)

 

'돌아보며'

⭐ 스택 자료구조는 재귀함수와 같이 DFS 그래프 알고리즘 구현 시 자주 쓰이니까 꼭 숙지해두자!