golang 写个栈(Stack)数据结构

2019-04-16  本文已影响0人  小风吹的我乱了

golang暂未有内置的Stack数据结构,工作需要就“借鉴”他人写了个

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
}

原理就是利用内置的list数据结构的PushBack,Remove等方法组合来实现进栈、出栈等操作
测试代码如下

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
上一篇下一篇

猜你喜欢

热点阅读