golang暫未有內(nèi)置的Stack數(shù)據(jù)結(jié)構(gòu),工作需要就“借鑒”他人寫了個
import "container/list"
type Stack struct {
list *list.List
}
func New() *Stack {
list := list.New()
return &Stack{
list: list,
}
}
func (p *Stack) Push(v interface{}) {
p.list.PushBack(v)
}
func (p *Stack) Pop() (interface{}, bool) {
e := p.list.Back()
if e != nil {
p.list.Remove(e)
return e.Value, true
}
return nil, false
}
func (p *Stack) Top() (interface{}, bool) {
e := p.list.Back()
if e != nil {
return e.Value, true
}
return nil, false
}
func (p *Stack) Len() int {
return p.list.Len()
}
func (p *Stack) Empty() bool {
return p.list.Len() == 0
}
原理就是利用內(nèi)置的list數(shù)據(jù)結(jié)構(gòu)的PushBack,Remove等方法組合來實現(xiàn)進棧、出棧等操作
測試代碼如下
func main() {
s := stack.New()
s.Push("hello")
s.Push("world")
v, ok := s.Pop()
fmt.Println(v, ok)
v, ok = s.Pop()
fmt.Println(v, ok)
v, ok = s.Pop()
fmt.Println(v, ok)
}
輸出:
world true
hello true
<nil> false