go---用两个栈实现队列

相信大家对这个题目都不陌生,中心思想就是,队列中包含两个栈,一个栈作为储存入队操作的,另一个栈作为储存出栈操作的,但是,在goland中,是没有栈这种数据结构的,那又有什么关系,goland还有万能的切片啊,所以我们先用切片实现栈,再用栈实现队列

代码:

package main

import "fmt"
//栈的结构
type stack struct{
	nums []int
}队列的结构
/
type queue struct {
	s1 *stack
	s2 *stack
}
//入栈操作
func (s *stack) push(n int){
	s.nums=append(s.nums,n)
}
//出栈操作
func (s *stack) pop() int{
	if s.isEmpty(){
		return -1
	}
	res:=s.nums[len(s.nums)-1]
	s.nums=s.nums[:len(s.nums)-1]
	return res
}
//判断栈是否为空
func (s *stack) isEmpty()bool{
	if len(s.nums)==0{
		return true
	}
	return false
}
//返回栈的长度
func (s *stack)len() int{
	return len(s.nums)
}
//初始话一个栈
func newStack()*stack{
	return &stack{
		nums: []int{},
	}
}
//初始化一个队列
func newQueue() queue{
	return queue{
		s1:newStack(),
		s2:newStack(),
	}
}
//入队操作
func (q *queue) Push(n int){
	q.s1.push(n)
}
//出队操作
func (q *queue)Pop() int{
	if q.s2.isEmpty(){
		for !q.s1.isEmpty(){
			q.s2.push(q.s1.pop())
		}
	}
	return q.s2.pop()
}
//返回队列中第一个值
func (q *queue) Get() int{
	if q.s2.isEmpty(){
		if q.s1.isEmpty(){
			return -1
		}
		return q.s1.nums[0]
	}
	return q.s2.nums[q.s2.len()-1]
}
//判断队列是否为空
func (q *queue) IsEmpty() bool{
	return q.s1.isEmpty() && q.s2.isEmpty()
}
func main(){
	q:=newQueue()
	q.Push(1)
	q.Push(2)
	q.Push(3)
	fmt.Println(q.Pop())
}
在这里插入代码片
C:\Users\my\go\src\hello\offer>go run stack-queue.go
1

注意只有当s2中没有元素的时候,才完成一次入栈操作,防止频繁操作栈

你可能感兴趣的:(go)