#4_ 파이썬으로 스택(Stack) 자료구조 구현하기

안녕하세요.

최근 파이썬을 공부하면서 기본적인 자료구조 알고리즘을 구현해보고자 생각이 들어서 포스팅을 하게 되었습니다.

제 노트북이 고장나서 노트북은 센터에 고이고이 잠들어있기 때문에.. 구름IDE로 코딩을 진행하였습니다.

나름 괜찮다고 생각이 드네요.

그럼 파이썬으로 구현한 스택 자료구조를 소개해드리도록 하겠습니다.

---

1. 스택이란?

먼저 스택 자료구조를 구현하기 전에 간단하게나마 해당 자료구조에 대해 알아보도록 하겠습니다. 

위의 그림과 같이 스택은 push와 pop이라는 기능을 가지고 있습니다.

스택 자료구조는 말 그대로, 쌓아 올리는 것과 같은 자료구조입니다.

즉, push은 item을 쌓아올리는 기능이고, pop은 쌓여져 있는 item에서 제일 위의 것을 꺼내는 작업입니다.

따라서 스택 자료구조는 Last In First Out(LIFO) 와 같은 특성을 가지고 있습니다.

2. 스택 구현하기

제가 파이썬 코드로써 구현해볼 것은 클래스로 스택개체를 만들고, push와 pop기능을 구현하고 해당 스택이 비어있는지 확인하는 isEmpty기능까지 구현합니다.

또한 스택이 비어있을때 pop기능을 시도하면 에러메세지를 출력하며 False 값을 리턴하게 하겠습니다.

완성된 코드를 보면 아래와 같습니다.

class stack:
    def __init__(self):
        self.stack_items=[]
    
    # pop 기능 구현
    def pop(self):
        item_length=len(self.stack_items)
        # 스택이 비어있을때는 에러메세지 출력
        if item_length < 1:
            print("Stack is empty!")
            return False
        # 가장 위에 있는 item을 반환하며 삭제
        result = self.stack_items[item_length-1]
        del self.stack_items[item_length-1]
        return result
    
    # push 기능 구현
    def push(self,x):
        self.stack_items.append(x)
        
    # isEmpty 기능 구현
    def isEmpty(self):
        return not self.stack_items
# 테스트 코드
test1 = stack()
print(test1.stack_items)
test1.push(3)
print(test1.stack_items)
test1.push(5)
test1.push(7)
test1.push(9)
test1.push(11)
print(test1.stack_items)
print(test1.isEmpty())
print(test1.pop())
print(test1.pop())
print(test1.pop())
print(test1.pop())
print(test1.pop())
print(test1.pop())
print(test1.isEmpty())

일부로 파이썬에서 기본적으로 제공하는 pop() 기능을 이용하지 않고 Stack 자료구조의 pop기능을 구현하였습니다.

추가적으로 궁금한 사항은 댓글에 남겨주세요 :)