Python棧

在英語詞典中，堆(Stack)表示將物件放在另一個物件上。 在這個資料結構中分配記憶體的方式是一樣的。 它以類似的方式儲存資料元素，類似在廚房中一堆盤子:一個在另一個之上存放。 所以堆疊資料資料允許操作的一端可以稱為棧頂。 可在棧頂上新增元素或僅從堆疊中移除元素。

在堆疊中，順序排列的最後一個元素將首先出現，因為只能從堆疊頂部移除。 這種功能稱為後進先出(LIFO)功能。 新增和刪除元素的操作稱為PUSH和POP。 在下面的程式中，我們將它實現為add和remove函式。首先宣告一個空列表並使用append()和pop()方法來新增和刪除資料元素。

推入堆疊

class Stack:

    def __init__(self):
        self.stack = []

    def add(self, dataval):
# Use list append method to add element
        if dataval not in self.stack:
            self.stack.append(dataval)
            return True
        else:
            return False
# Use peek to look at the top of the stack

    def peek(self):     
        return self.stack[0]

AStack = Stack()
AStack.add("Mon")
AStack.add("Tue")
AStack.peek()
print(AStack.peek())
AStack.add("Wed")
AStack.add("Thu")
print(AStack.peek())

執行上面範例程式碼，得到以下結果 -

Mon
Mon

堆疊移除

只能從堆疊中移除資料元素，下面實現了一個可以實現這一功能的python程式。 以下程式中的remove函式返回最上面的元素。 首先通過計算堆疊的大小來檢查頂層元素，然後使用內建的pop()方法找出最頂層的元素。參考以下程式碼實現 -

class Stack:

    def __init__(self):
        self.stack = []

    def add(self, dataval):
# Use list append method to add element
        if dataval not in self.stack:
            self.stack.append(dataval)
            return True
        else:
            return False

# Use list pop method to remove element
    def remove(self):
        if len(self.stack) <= 0:
            return ("No element in the Stack")
        else:
            return self.stack.pop()

AStack = Stack()
AStack.add("Mon")
AStack.add("Tue")
print(AStack.remove())
AStack.add("Wed")
AStack.add("Thu")
print(AStack.remove())

執行上面範例程式碼，得到以下結果 -

Tue
Thu