iOS 開發中用到的 UINavigationController 其實就是一個棧結構,可以存放多個子控制器,棧頂控制器為當前顯示的控制器,棧底控制器為 rootViewController ,當 push 操作時會將一個新的 UIViewController 存放到棧中去, pop 操作時會將棧頂控制器從 UINavigationController中移除

Github代碼地址 : https://github.com/ZhaoBingDong/DataStructure.git

今天用 swift 代碼來實現一個棧結構,使其具備一下特點
1 能夠插入一個新的元素到棧中
2 能夠從棧頂取出一個元素
3 能夠清空棧中的所有元素
4 能夠遍歷棧中所有元素
5 當棧為空或者滿的時候不進行插入和取出的任何操作.

Simulator Screen Shot 2017年3月13日 下午1.39.51.png

一 如何創建這個棧 MyStack

   init(_ capacity : Int) {
        stackCapacity = capacity;
        clearStacks()    
   }
    /// 清空棧所有元素
    open func clearStacks() {
        self.stacks = [T]()
        stacklength = 0
        stackTop    = 0
    }
    

二 判斷棧長度和棧是否為空,為滿

    // 獲取棧元素個數
      private var stacklength : Int = 0

    /// 棧是否為空
    open func stackEmpty() -> Bool {
        return stacklength == 0
    }
    
    /// 棧是否為滿
    open func stackFull() -> Bool {
        return stacklength == stackCapacity
    }

三 往棧中插入一個元素.

    
    /// 入棧一個元素
   public func push(_ element : String) -> Bool  {
          if stackFull() {
              return false
          } else {
            self.stacks?.insert(element, at: stackTop)
            stackTop+=1;
            stacklength+=1;
           return true
        }
   }

/// 從棧頂取出一個元素

    public func pop() -> T? {
        if stackEmpty() {
            return nil;
        } else {
            stackTop-=1;
            stacklength-=1;
            let string = self.stacks?[stackTop]
            print("出棧一個元素\(String(describing: string))")
            return string
        }
    }

四 遍歷棧中元素 , 用來驗證棧中插入 清空 取出元素后棧中元素的變化情況

        /// 遍歷棧中所有元素
   open func stackTraverse() {
        
        if stackEmpty() {
            print("棧中沒有元素");
            return
        }
        
        for i in 0..<stackTop {            
            print("\n\((self.stacks?[i])!),\n")
        }
        
 }

結合代碼實現了一個帶 UI 界面的棧操作示意圖

Untitled.gif

邏輯代碼如下 :

    // 清空棧操作
    @IBAction func clearStacks(_ sender: UIButton) {
        
        for view in self.contentView.subviews {
            view.removeFromSuperview()
        }
        
        self.stack?.clearStacks()
        
    }
   // 出棧操作
    @IBAction func pop(_ sender: UIButton) {
        
        if (self.stack?.stackEmpty())! {
            return
        }
        
        // 根據棧頂索引 取出棧頂的 label
        let topLabel = self.contentView.subviews[((self.stack?.stackTop)!-1)]
        topLabel.removeFromSuperview()
        // 出棧一個元素
        
        let _ = self.stack?.pop()
        
    }
    // 入棧操作
    @IBAction func push(_ sender: UIButton) {
    
        if (self.stack?.stackFull())! {
            return
        }
        
        let array : [String] = ["大兵布萊恩特","張三","李四","科比","詹姆斯","杜蘭特","庫里"]
        let index = arc4random_uniform(UInt32(array.count))
        let name  = array[Int(index)]
        let topLabel = self.contentView.subviews.last as? UILabel
        
        var labelY  : CGFloat  = 0.0
        if topLabel != nil {
            labelY   = (topLabel?.frame.minY)!-100.0;
        } else {
            labelY   = 300.0
        }
        
        let label = UILabel(frame: CGRect(x: 0.0, y: labelY, width: self.contentView.frame.size.width, height: 100.0))
        label.textColor = UIColor.white
        label.textAlignment = .center
        label.text = "view\((self.stack?.stackTop)!) name = \(name)"
        label.tag  = (self.stack?.stackTop)!
        label.backgroundColor = UIColor.RandomColor()
        
        self.contentView.addSubview(label)
        
        // 入棧一個元素
        
        let _ = self.stack?.push(name)
        
    }