node通過prev字段進行關聯,stack維護棧頂節點
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