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