stack

Author: xiaomeng79

data-structures/README.md

@@ -6,6 +6,10 @@
 
 [算法时间复杂度的计算](http://univasity.iteye.com/blog/1164707)
 
+[go实现01](https://github.com/arnauddri/algorithms)
+
+[go实现02](https://github.com/floyernick/Data-Structures-and-Algorithms)
+
 ### 逻辑结构:
 
 具体问题抽象出来的数学模型

data-structures/stack/stack.go

@@ -0,0 +1,60 @@
+package stack
+
+import (
+	"errors"
+	"sync"
+)
+
+const ARRAY_SIZE = 10
+
+//数组实现
+type Stack struct {
+	data [ARRAY_SIZE]int
+	top  int
+	lock sync.Mutex
+}
+
+func New() *Stack {
+	return &Stack{}
+}
+
+func (s *Stack) Len() int {
+	s.lock.Lock()
+	defer s.lock.Unlock()
+	return s.top
+}
+
+func (s *Stack) IsEmpty() bool {
+	s.lock.Lock()
+	defer s.lock.Unlock()
+	return s.top == 0
+}
+
+func (s *Stack) Push(i int) error {
+	s.lock.Lock()
+	defer s.lock.Unlock()
+	if s.top == ARRAY_SIZE {
+		return errors.New("栈已满")
+	}
+	s.data[s.top] = i
+	s.top++
+	return nil
+}
+func (s *Stack) Pop() (int, error) {
+	s.lock.Lock()
+	defer s.lock.Unlock()
+	if s.top == 0 {
+		return 0, errors.New("栈空")
+	}
+	s.top--
+	return s.data[s.top], nil
+}
+
+func (s *Stack) Peek() (int, error) {
+	s.lock.Lock()
+	defer s.lock.Unlock()
+	if s.top == 0 {
+		return 0, errors.New("栈空")
+	}
+	return s.data[s.top-1], nil
+}

data-structures/stack/stack_array.go

@@ -0,0 +1,35 @@
+package stack
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestNew(t *testing.T) {
+	s := New()
+	for i := 0; i < 10; i++ {
+		err := s.Push(i)
+		assert.NoError(t, err)
+	}
+	err := s.Push(11)
+	assert.Error(t, err)
+	for i := 9; i >= 0; i-- {
+		k, _ := s.Pop()
+		assert.Equal(t, i, k)
+	}
+	_, err = s.Pop()
+	assert.Error(t, err)
+}
+
+func TestNewListStack(t *testing.T) {
+	s := NewListStack()
+	for i := 0; i < 10; i++ {
+		s.Push(i)
+	}
+	for i := 9; i >= 0; i-- {
+		k, _ := s.Pop()
+		assert.Equal(t, i, k)
+	}
+	_, err := s.Pop()
+	assert.Error(t, err)
+}

data-structures/stack/stack_array_test.go

@@ -0,0 +1,54 @@
+package stack
+
+import "errors"
+
+//使用链表实现栈
+//以下是使用单向链表，使用头插法实现，也可以使用标准库的双向链表实现
+//节点，需要动态申请内存空间
+type Node struct {
+	data int
+	next *Node
+}
+
+//栈
+type ListStack struct {
+	top *Node
+}
+
+func NewListStack() *ListStack {
+	return &ListStack{}
+}
+
+//是否为空
+func (ls *ListStack) IsEmpty() bool {
+	return ls.top == nil
+}
+
+//入栈,头插法
+func (ls *ListStack) Push(i int) {
+	//新建节点
+	n := &Node{i, nil}
+	if ls.top != nil {
+		n.next = ls.top
+	}
+	ls.top = n
+}
+
+//出栈,从头弹出
+func (ls *ListStack) Pop() (int, error) {
+	if ls.top == nil {
+		return 0, errors.New("栈空")
+	}
+	i := ls.top.data
+	ls.top = ls.top.next
+	return i, nil
+
+}
+
+//查看
+func (ls *ListStack) Peek() (int, error) {
+	if ls.top == nil {
+		return 0, errors.New("栈空")
+	}
+	return ls.top.data, nil
+}