相信大家对这个题目都不陌生,中心思想就是,队列中包含两个栈,一个栈作为储存入队操作的,另一个栈作为储存出栈操作的,但是,在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中没有元素的时候,才完成一次入栈操作,防止频繁操作栈