Python數據結構——棧

用兩種數據結構實現棧
1、使用list,代碼如下:

class Stack :
  # 初始化一個空的棧
  def __init__( self ):
    self._theItems = list()

  #返回True或False.
  def isEmpty( self ):
    return len( self ) == 0

  # 返回棧中元素的個數
  def __len__ ( self ):
    return len( self._theItems )

 #返回棧頂元素
  def peek( self ):
    assert not self.isEmpty(), "棧不能爲空"
    return self._theItems[-1]

 # 刪除並返回棧頂元素
  def pop( self ):
    assert not self.isEmpty(), "無法從空棧刪除一個元素"
    return self._theItems.pop()

 # 讓一個元素入棧
  def push( self, item ):
    self._theItems.append( item )

它的使用及控制檯輸出:

stack = Stack()
stack.push(123)
stack.push("PHP")
stack.push("c語言")
print(len(stack))
print(stack.peek())
print(stack.pop())
print(len(stack))

這裏寫圖片描述

2、使用鏈表實現棧,代碼如下:

class Stack2 :
 # 創建一個空棧.
  def __init__( self ):
   self._top = None
   self._size = 0

# 如果棧空返回True否則返回False 
  def isEmpty( self ):
    return self._top is None

 # 返回棧中元素數量
  def __len__( self ):
    return self._size

 # 返回棧中頂部元素.
  def peek( self ):
    assert not self.isEmpty(), "Cannot peek at an empty stack"
    return self._top.item

 # 刪除並返回棧頂部元素
  def pop( self ):
    assert not self.isEmpty(), "Cannot pop from an empty stack"
    node = self._top
    self.top = self._top.next
    self._size -= 1
    return node.item

 # 把一個元素推入棧頂
  def push( self, item ) :
    self._top = _StackNode( item, self._top )
    self._size += 1

 # 創建棧節點的私有存儲類
class _StackNode :
 def __init__( self, item, link ) :
   self.item = item
   self.next = link

Stack2類的構造器爲每個棧創建了兩個實例變量。top用來保存鏈表,size跟蹤棧中元素的數量。
StackNode 類創建鏈表節點

Stack2類的使用及控制檯輸出如下:

stack2 = Stack2()
stack2.push(199)
stack2.push("apple")
stack2.push("香蕉")
print(len(stack2))
print(stack2.peek())
print(stack2.pop())
print(len(stack2))

這裏寫圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章