2. Golang 实现

// 栈信息
type Stack struct {
    list *SingleList
}

// Init 初始化栈
func (s *Stack) Init()  {
    s.list = new(SingleList)
    s.list.Init()
}

 

2.3. 压入栈（push)

往栈内插入数据，称为push，在这里对于栈的压入压出都是对链表的表头处理，当然也可以对表尾处理，道理都是一样的

 

// Push 压入栈
func (s *Stack)Push(data interface{}) bool {
    node := &SingleNode{
        Data: data,
    }
    return s.list.Insert(0, node)
}

 

2.4. 压出栈(pop)

取出栈顶数据，称为pop。

// Pop 压出栈
func (s *Stack)Pop() interface{}{
    node := s.list.Get(0)
    if node != nil {
        s.list.Delete(0)
        return node.Data
    }
    return nil
}

 

2.5. 查看栈顶数据(peek)

只查看栈顶元素，并不取出

// Peek 查看栈顶元素
func (s *Stack)Peek() interface{}{
    node := s.list.Get(0)
    if node != nil {
        return node.Data
    }
    return nil
}

2.6. 栈长度(size)

查询栈当前元素数量

// Size 获取栈的长度
func (s *Stack)Size()uint{
    return s.list.Size
}

3. 单元测试

package dstr

import(
    "testing"
)

func TestStack_Init(t *testing.T)  {
    stack := new(Stack)
    stack.Init()
    t.Log("stack init success")
}

func TestStack_Push(t *testing.T){
    stack := new(Stack)
    stack.Init()
    b := stack.Push(1)
    if !b {
        t.Error("stack push failed")
        return
    }
    t.Log("stack push success")
    data := stack.Peek()
    var (
        ok bool
        num int
    )
    if num, ok = data.(int); ok && num == 1{
        t.Log("stack push and peek success")
        return
    }
    t.Error("stack push success but peek failed")
}

func TestStack_Pop(t *testing.T){
    stack := new(Stack)
    stack.Init()
    d1 := stack.Pop()
    if d1 != nil{
        t.Error("empty stack pop error")
        return
    }
    t.Log("empty stack pop success")

    stack.Push(1)
    stack.Push(2)
    stack.Push(3)
    d2 := stack.Pop()
    var (
        ok bool
        num int
    )
    if num, ok = d2.(int); ok && num == 3 && stack.Size() == 2{
        t.Log("stack pop success")
        return
    }
    t.Error("stack pop failed")
}

func TestStack_Peek(t *testing.T){
    stack := new(Stack)
    stack.Init()
    d := stack.Peek()
    if d == nil {
        t.Log("empty stack peek success")
        return
    }
    t.Error("empty stack peek fail")
}