fix bugs of array stack

Jentlezhi · Oct 10, 2018 · a9dff3d · a9dff3d
1 parent e2b841f
commit a9dff3d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/go/08_stack/StackBasedOnArray.go b/go/08_stack/StackBasedOnArray.go
@@ -28,12 +28,17 @@ func (this *ArrayStack) IsEmpty() bool {
 }
 
 func (this *ArrayStack) Push(v interface{}) {
-	this.data = append(this.data, v)
 	if this.top < 0 {
 		this.top = 0
 	} else {
 		this.top += 1
 	}
+
+	if this.top > len(this.data)-1 {
+		this.data = append(this.data, v)
+	} else {
+		this.data[this.top] = v
+	}
 }
 
 func (this *ArrayStack) Pop() interface{} {

diff --git a/go/08_stack/StackBasedOnArray_test.go b/go/08_stack/StackBasedOnArray_test.go
@@ -6,7 +6,12 @@ func TestArrayStack_Push(t *testing.T) {
 	s := NewArrayStack()
 	s.Push(1)
 	s.Push(2)
+	t.Log(s.Pop())
 	s.Push(3)
+	t.Log(s.Pop())
+	t.Log(s.Pop())
+	s.Push(4)
+	t.Log(s.Pop())
 	s.Print()
 }