完成stack的接口标准

stack/liststack/stack.go

@@ -0,0 +1,102 @@
+package lastack
+
+import (
+	"github.com/474420502/focus/stack"
+	"github.com/davecgh/go-spew/spew"
+)
+
+type Node struct {
+	value interface{}
+	down  *Node
+}
+
+type Stack struct {
+	top  *Node
+	size uint
+}
+
+func assertImplementation() {
+	var _ stack.IStack = (*Stack)(nil)
+}
+
+
+func New() *Stack {
+	s := &Stack{}
+	s.size = 0
+	return s
+}
+
+func (as *Stack) Clear() {
+	as.size = 0
+	as.top = nil
+}
+
+func (as *Stack) Empty() bool {
+	return as.size == 0
+}
+
+func (as *Stack) Size() uint {
+	return as.size
+}
+
+// String 从左到右 左边第一个表示Top 如链表 a(top)->b->c
+func (as *Stack) String() string {
+	content := ""
+	cur := as.top
+	for ; cur != nil; cur = cur.down {
+		content += spew.Sprint(cur.value) + " "
+	}
+
+	if len(content) > 0 {
+		content = content[0 : len(content)-1]
+	} else {
+		content = ""
+	}
+
+	return content
+}
+
+func (as *Stack) Values() []interface{} {
+
+	if as.size == 0 {
+		return nil
+	}
+
+	result := make([]interface{}, as.size, as.size)
+
+	cur := as.top
+	n := 0
+	for ; cur != nil; cur = cur.down {
+		result[n] = cur.value
+		n++
+	}
+
+	return result
+}
+
+func (as *Stack) Push(v interface{}) {
+	nv := &Node{value: v}
+	nv.down = as.top
+	as.top = nv
+	as.size++
+}
+
+func (as *Stack) Pop() (interface{}, bool) {
+	if as.size == 0 {
+		return nil, false
+	}
+
+	as.size--
+
+	result := as.top
+	as.top = as.top.down
+	result.down = nil
+	return result.value, true
+}
+
+func (as *Stack) Peek() (interface{}, bool) {
+	if as.size == 0 {
+		return nil, false
+	}
+	return as.top.value, true
+}

stack/liststack/stack_test.go

@@ -0,0 +1,182 @@
+package lastack
+
+import "testing"
+
+func TestBase(t *testing.T) {
+	s := New()
+	if !s.Empty() {
+		t.Error("stack not empty")
+	}
+
+	if s.Size() != 0 {
+		t.Error("size != 0")
+	}
+
+	if s.Values() != nil {
+		t.Error(s.Values())
+	}
+}
+
+func TestPush(t *testing.T) {
+	s := New()
+
+	for i := 0; i < 5; i++ {
+		s.Push(i)
+	}
+
+	if s.Empty() {
+		t.Error("stack is empty")
+	}
+
+	if s.Size() == 0 {
+		t.Error("size == 0")
+	}
+
+	if s.Values() == nil {
+		t.Error("Values() != nil")
+	}
+
+	if s.String() != "4 3 2 1 0" {
+		t.Error(s.String())
+	}
+
+	if v, ok := s.Peek(); ok {
+		if v != 4 {
+			t.Error("why top != 4")
+		}
+	} else {
+		t.Error("not ok")
+	}
+
+	if v, ok := s.Pop(); ok {
+		if v != 4 {
+			t.Error("why top != 4")
+		}
+	} else {
+		t.Error("not ok")
+	}
+
+	if s.Size() != 4 {
+		t.Error("pop a element, size: 5 - 1 = 4")
+	}
+
+	//
+	if v, ok := s.Pop(); ok {
+		if v != 3 {
+			t.Error("why top != 3")
+		}
+	} else {
+		t.Error("not ok")
+	}
+
+	if s.Size() != 3 {
+		t.Error("pop a element, size: 4 - 1 = 3")
+	}
+
+	for _, ok := s.Pop(); ok != false; _, ok = s.Pop() {
+
+	}
+
+	if !s.Empty() && s.Size() != 0 {
+		t.Error("pop all, stack should be empty")
+	}
+
+	for i := 0; i < 5; i++ {
+		s.Push(i)
+	}
+
+	if s.Size() != 5 {
+		t.Error("size != 5")
+	}
+
+	s.Clear()
+	if !s.Empty() && s.Size() != 0 {
+		t.Error("pop all, stack should be empty")
+	}
+
+	if v, ok := s.Peek(); v != nil || ok != false {
+		t.Error("should be v == nil and ok == false")
+	}
+}
+
+// func BenchmarkPush(b *testing.B) {
+// 	s := New()
+// 	b.N = 200000
+// 	for i := 0; i < b.N; i++ {
+// 		v := randomdata.Number(0, 65535)
+// 		s.Push(v)
+// 	}
+// }
+
+// func BenchmarkGodsPush(b *testing.B) {
+// 	s := arraystack.New()
+// 	b.N = 200000
+// 	for i := 0; i < b.N; i++ {
+// 		v := randomdata.Number(0, 65535)
+// 		s.Push(v)
+// 	}
+// }
+
+// func BenchmarkPop(b *testing.B) {
+// 	s := New()
+// 	b.N = 200000
+
+// 	for i := 0; i < b.N; i++ {
+// 		v := randomdata.Number(0, 65535)
+// 		s.Push(v)
+// 	}
+
+// 	b.ResetTimer()
+// 	b.StartTimer()
+
+// 	for i := 0; i < b.N; i++ {
+// 		s.Pop()
+// 	}
+// }
+
+// func BenchmarkGodsPop(b *testing.B) {
+// 	s := arraystack.New()
+// 	b.N = 200000
+
+// 	for i := 0; i < b.N; i++ {
+// 		v := randomdata.Number(0, 65535)
+// 		s.Push(v)
+// 	}
+
+// 	b.ResetTimer()
+// 	b.StartTimer()
+
+// 	for i := 0; i < b.N; i++ {
+// 		s.Pop()
+// 	}
+// }
+
+// func BenchmarkValues(b *testing.B) {
+// 	s := New()
+// 	for i := 0; i < b.N; i++ {
+// 		v := randomdata.Number(0, 65535)
+// 		s.Push(v)
+// 	}
+
+// 	b.ResetTimer()
+// 	b.StartTimer()
+
+// 	for i := 0; i < b.N; i++ {
+// 		s.Values()
+// 	}
+// }
+
+// func BenchmarkGodsValues(b *testing.B) {
+// 	s := arraystack.New()
+// 	for i := 0; i < b.N; i++ {
+// 		v := randomdata.Number(0, 65535)
+// 		s.Push(v)
+// 	}
+
+// 	b.ResetTimer()
+// 	b.StartTimer()
+
+// 	for i := 0; i < b.N; i++ {
+// 		s.Values()
+// 	}
+// }