go fmt for stack

pull/7/head
Alexander Kuleshov 7 years ago
parent a4d6be93c7
commit 307fadc6e2
No known key found for this signature in database
GPG Key ID: 11DB55FD14934DB6

@ -1,61 +1,61 @@
package stack package stack
type Stack struct { type Stack struct {
st []interface{} st []interface{}
len int len int
} }
func New() *Stack { func New() *Stack {
stack := &Stack{} stack := &Stack{}
stack.st = make([]interface{}, 1) stack.st = make([]interface{}, 1)
stack.len = 0 stack.len = 0
return stack return stack
} }
func (stack *Stack) Length() int { func (stack *Stack) Length() int {
return stack.len return stack.len
} }
func (stack *Stack) Pop() { func (stack *Stack) Pop() {
stack.st = stack.st[1:] stack.st = stack.st[1:]
stack.len -= 1 stack.len -= 1
} }
func (stack *Stack) Peek() interface{} { func (stack *Stack) Peek() interface{} {
return stack.st[0] return stack.st[0]
} }
func (stack *Stack) IsEmpty() bool { func (stack *Stack) IsEmpty() bool {
return (stack.len == 0) return (stack.len == 0)
} }
func (stack *Stack) Push(value interface{}) { func (stack *Stack) Push(value interface{}) {
add(stack, value) add(stack, value)
} }
func add(slice *Stack, value interface{}) { func add(slice *Stack, value interface{}) {
slice.len += 1 slice.len += 1
var tmpSlice []interface{} = make([]interface{}, slice.len) var tmpSlice []interface{} = make([]interface{}, slice.len)
if slice.len == 0 { if slice.len == 0 {
slice.st[0] = value slice.st[0] = value
return return
} }
for i:=0; i < slice.len; i++ { for i := 0; i < slice.len; i++ {
tmpSlice[i] = 0 tmpSlice[i] = 0
} }
for i:=0; i < slice.len; i++ { for i := 0; i < slice.len; i++ {
if i == 0 { if i == 0 {
tmpSlice[0] = value tmpSlice[0] = value
} else { } else {
tmpSlice[i] = slice.st[i - 1] tmpSlice[i] = slice.st[i-1]
} }
if i == slice.len - 1 { if i == slice.len-1 {
break break
} }
} }
slice.st = tmpSlice slice.st = tmpSlice
} }

@ -3,23 +3,23 @@ package stack
import "testing" import "testing"
func Test_Stack(t *testing.T) { func Test_Stack(t *testing.T) {
stack := New() stack := New()
stack.Push(5) stack.Push(5)
stack.Push(6) stack.Push(6)
stack.Push(7) stack.Push(7)
if stack.Length() != 3 {
t.Error("[Error] stack length is wrong")
}
stack.Pop() if stack.Length() != 3 {
t.Error("[Error] stack length is wrong")
}
if stack.Length() != 2 { stack.Pop()
t.Error("[Error] stack length is wrong after pop")
} if stack.Length() != 2 {
t.Error("[Error] stack length is wrong after pop")
if stack.Peek() != 6 { }
t.Error("[Error] stack Peek is wrong")
} if stack.Peek() != 6 {
t.Error("[Error] stack Peek is wrong")
}
} }

Loading…
Cancel
Save