💡 스택

이미지 출처

스택이란?

스택(stack)이란 쌓다라는 의미로 데이터를 위로 차곡차곡 쌓아 올린 형태의 자료 구조이다.
가장 먼저 삽입된 데이터가 맨 마지막에 삭제, 즉 가장 나중에 들어온 데이터가 가장 먼저 나가는 LIFO 구조이다.

📢10828번 문제

백준 문제 링크

📢풀이

사용언어 : python

command :은 명령을 입력받는 변수로 입력된 command에 따라 명령을 수행한다.
if 문을 사용하여 command에 입력된 값이 push, pop, size, empty , top일 경우에 따라 각각의 명령을 수행한다.

🔎 코드

import sys
N = int(input())
stack = []
for i in range(N):
    command = sys.stdin.readline().split()
    if command[0] == "push":
        stack.append(int(command[1]))
    elif command[0] == "pop":
        if len(stack) > 0:
            print(stack.pop())
        else:
            print(-1)
    elif command[0] == "size":
        print(len(stack))
    elif command[0] == "empty":
        if not stack:
            print(1)
        else:
            print(0)
    elif command[0] == "top":
        if not stack:
            print(-1)
        else:
            print(stack[-1])