Go入门 - stack包

最新推荐文章于 2024-08-23 07:33:49 发布

DDDDemo

最新推荐文章于 2024-08-23 07:33:49 发布

阅读量6.1k

点赞数

分类专栏： Go

Go 专栏收录该内容

31 篇文章 2 订阅

订阅专栏

1、建立一个独立的包，为stack的实现创建一个合适的包，Push Pop和Stack类型需要被导出。并为此包编写一个单元测试，至少测试Push后Pop的工作情况。

stack-as-package.go文件

package stack

type Stack struct {
	i 	 int
	data [10]int
}

func (s *Stack) Push(k int)  {
	s.data[s.i] = k
	s.i++
}

func (s *Stack) Pop() (ret int) {
		s.i--
		ret = s.data[s.i]
		return
}

pushpop_test.go

package stack

import "testing"

func TestPushPop(t *testing.T)  {
	c := new(Stack)
	c.Push(5)
	if c.Pop() != 5 {
		t.Log("Pop doesn't give 5")
		t.Fail()
	}
}

依次执行命令

> mkdir $GOPATH/src/stack

> cp pushpop_test.go $GOPATH/src/stack

> cp stack-as-package.go $GOPATH/src/stack

> go test stack

2、使用stack包创建逆波兰计算器

package main

import (
	"bufio"
	"fmt"
	"os"
	"strconv"
)

var reader *bufio.Reader = bufio.NewReader(os.Stdin)
var st = new(Stack)

type Stack struct {
	i int
	data [10]int
}

func (s *Stack) Push(k int)  {
	if s.i +1 >9 {
		return
	}
	s.data[s.i] = k
	s.i++
}

func (s *Stack) Pop() (ret int) {
	s.i--
	if s.i < 0 {
		s.i = 0
		return
	}
	ret = s.data[s.i]
	return
}

func main()  {
	for {
		s, err := reader.ReadString('\n')
		var token string
		if err != nil {
			return
		}
		for _,c := range s {
			switch {
				case c > '0' && c < '9' :
					token = token + string(c)
				case c == ' ':
					r,_ := strconv.Atoi(token)
					st.Push(r)
					token = ""
				case c == '+':
					fmt.Printf("%d\n",st.Pop() + st.Pop())
				case c == '*':
					fmt.Printf("%d\n",st.Pop() * st.Pop())
				case c == '-':
					fmt.Printf("%d\n",st.Pop() - st.Pop())
				case c == 'q':
					return
				default:
					//error
			}

		}
	}
}

如程序所示：输入数字后输入空格，此数字才会被写入到栈中