golang数据结构之队列

package queue

import (
    "errors"
    "fmt"
)

//Queue 使用结构体管理队列
type Queue struct {
    MaxSize int
    Array   [4]int //模拟队列
    Front   int    //队列首位
    Rear    int    //队列尾部
}

//AddQueue 向队列中添加一个值
func (q *Queue) AddQueue(val int) (err error) {
    q.MaxSize = 4
    //先判断队列是否已满
    if q.Rear == q.MaxSize-1 {
        return errors.New("队列已满")
    }
    q.Rear++
    q.Array[q.Rear] = val
    return
}

//GetQueue 得到一个值
func (q *Queue) GetQueue() (val int, err error) {
    if q.Front == q.Rear {
        return -1, errors.New("队列已空")
    }
    q.Front++
    val = q.Array[q.Front]
    return val, err
}

//ShowQueue 显示队列
func (q *Queue) ShowQueue() {
    for i := q.Front + 1; i <= q.Rear; i++ {
        fmt.Printf("queue[%d]=%v\t", i, q.Array[i])
    }
}

package main

import (
    "fmt"
    "go_code/data_structure/queue"
    "os"
)

func main() {
    var key string
    var val int
    q := &queue.Queue{
        MaxSize: 4,
        Front:   -1,
        Rear:    -1,
    }
    for {
        fmt.Println("------------------------------")
        fmt.Println("1.输入add表示添加数据到队列")
        fmt.Println("2.输入get表示从队列中获取数据")
        fmt.Println("3.输入show表示显示队列")
        fmt.Println("4.输入exit表示退出")
        fmt.Println("------------------------------")
        fmt.Scanln(&key)
        switch key {
        case "add":
            fmt.Println("请输入要添加的值：")
            fmt.Scanln(&val)
            err := q.AddQueue(val)
            if err != nil {
                fmt.Println(err)
            } else {
                fmt.Println("添加成功")
                fmt.Println("Rear：", q.Rear)
            }
        case "get":
            val, err := q.GetQueue()
            if err != nil {
                fmt.Println(err)
            } else {
                fmt.Println("得到的值为：", val)
                fmt.Println("Rear：", q.Front)
            }

        case "show":
            q.ShowQueue()
            fmt.Println()
        case "exit":
            os.Exit(0)
        }
    }
}