golang中用鏈表實現棧

node通過prev字段進行關聯,stack維護棧頂節點
golang中用鏈表實現棧

stack.go

package stack

type Stack struct {
    top    *node
    length int
}

type node struct {
    value interface{}
    prev  *node
}

// 創建一個棧
func New() *Stack {
    return &Stack{nil, 0}
}

// 取棧長度
func (s *Stack) Len() int {
    return s.length
}

// 查看棧頂元素
func (s *Stack) Peek() interface{} {
    if s.length == 0 {
        return nil
    }
    return s.top.value
}

// 出棧
func (s *Stack) Pop() interface{} {
    if s.length == 0 {
        return nil
    }
    n := s.top
    s.top = n.prev
    s.length--
    return n.value
}

// 入棧
func (s *Stack) Push(value interface{}) {
    n := &node{value, s.top}
    s.top = n
    s.length++
}

main.go

package main

import (
    "./stack"
    "fmt"
)

func main() {
    st := stack.New()
    st.Push(111)
    st.Push(222)

    fmt.Println(st.Peek(), st.Len())

    item := st.Pop()
    fmt.Println(item)

    fmt.Println(st.Peek(), st.Len())
}

輸出:

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