登录
首页 >  Golang >  Go问答

如何使用golang实现一个栈和队列?

来源:Golang技术栈

时间:2023-03-05 14:13:40 315浏览 收藏

各位小伙伴们,大家好呀!看看今天我又给各位带来了什么文章?本文标题《如何使用golang实现一个栈和队列?》,很明显是关于Golang的文章哈哈哈,其中内容主要会涉及到golang、queue、stack等等,如果能帮到你,觉得很不错的话,欢迎各位多多点评和分享!

问题内容

如何使用golang实现一个栈和队列?

正确答案

可以像下面这样

package main

import (
	"fmt"
)

// 节点
type Node struct {
	Value int
}

func (n *Node) String() string {
	return fmt.Sprint(n.Value)
}

// 实例化栈
func NewStack() *Stack {
	return &Stack{}
}

// 栈结构
type Stack struct {
	nodes []*Node
	count int
}

// 入栈
func (s *Stack) Push(n *Node) {
	s.nodes = append(s.nodes[:s.count], n)
	s.count++
}

// 出栈
func (s *Stack) Pop() *Node {
	if s.count == 0 {
		return nil
	}
	s.count--
	return s.nodes[s.count]
}

// 新建队列
func NewQueue(size int) *Queue {
	return &Queue{
		nodes: make([]*Node, size),
		size:  size,
	}
}

//队列结构
type Queue struct {
	nodes []*Node
	size  int
	head  int
	tail  int
	count int
}

// 入队
func (q *Queue) Push(n *Node) {
	if q.head == q.tail && q.count > 0 {
		nodes := make([]*Node, len(q.nodes)+q.size)
		copy(nodes, q.nodes[q.head:])
		copy(nodes[len(q.nodes)-q.head:], q.nodes[:q.head])
		q.head = 0
		q.tail = len(q.nodes)
		q.nodes = nodes
	}
	q.nodes[q.tail] = n
	q.tail = (q.tail + 1) % len(q.nodes)
	q.count++
}

// 出队
func (q *Queue) Pop() *Node {
	if q.count == 0 {
		return nil
	}
	node := q.nodes[q.head]
	q.head = (q.head + 1) % len(q.nodes)
	q.count--
	return node
}

func main() {
	// 测试栈
	s := NewStack()
	s.Push(&Node{3})
	s.Push(&Node{5})
	s.Push(&Node{7})
	s.Push(&Node{9})
	fmt.Println(s.Pop(), s.Pop(), s.Pop(), s.Pop())

	// 测试队列
	q := NewQueue(1)
	q.Push(&Node{2})
	q.Push(&Node{4})
	q.Push(&Node{6})
	q.Push(&Node{8})
	fmt.Println(q.Pop(), q.Pop(), q.Pop(), q.Pop())
}

运行结果

9 7 5 3
2 4 6 8

今天带大家了解了golang、queue、stack的相关知识,希望对你有所帮助;关于Golang的技术知识我们会一点点深入介绍,欢迎大家关注golang学习网公众号,一起学习编程~

声明:本文转载于:Golang技术栈 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>
评论列表