问题:如何用栈实现队列?如何用队列实现栈?
原题对应225. 用队列实现栈 - 力扣(LeetCode以及232. 用栈实现队列 - 力扣(LeetCode),非常遗憾的是,golang没有专门的栈和队列的数据结构。只能通过切片来代表栈(先进后出)、队列(先进先出)。
代码如下:
栈实现队列:使用两个栈,push时,向栈1放元素;pop时,先将栈1元素出栈再向栈2入栈。
//构造MyQueue,使用两个栈,一个进栈,一个出栈
type MyQueue struct {
inStack, outStack []int
}
func Constructor() MyQueue {
return MyQueue{}
}
func (q *MyQueue) Push(x int) {
q.inStack = append(q.inStack, x)
}
func (q *MyQueue) in2out() {
for len(q.inStack) > 0 {
q.outStack = append(q.outStack, q.inStack[len(q.inStack)-1])
q.inStack = q.inStack[:len(q.inStack)-1]
}
}
func (q *MyQueue) Pop() int {
if len(q.outStack) == 0 {
q.in2out()
}
x := q.outStack[len(q.outStack)-1]
q.outStack = q.outStack[:len(q.outStack)-1]
return x
}
func (q *MyQueue) Peek() int {
if len(q.outStack) == 0 {
q.in2out()
}
return q.outStack[len(q.outStack)-1]
}
func (q *MyQueue) Empty() bool {
return len(q.inStack) == 0 && len(q.outStack) == 0
}
队列实现栈:使用两个队列,push向队列2push数据,再把队列1中数据送入队列2中,这样使得刚刚送入队列的元素排在最前面,pop时候直接送出即可。
type MyStack struct {
queue1, queue2 []int
}
func Constructor() (s MyStack) {
return MyStack{}
}
func (s *MyStack) Push(x int) {
s.queue2 = append(s.queue2, x)
for len(s.queue1) > 0 {
s.queue2 = append(s.queue2, s.queue1[0])
s.queue1 = s.queue1[1:]
}
s.queue1, s.queue2 = s.queue2, s.queue1
}
func (s *MyStack) Pop() int {
v := s.queue1[0]
s.queue1 = s.queue1[1:]
return v
}
func (s *MyStack) Top() int {
return s.queue1[0]
}
func (s *MyStack) Empty() bool {
return len(s.queue1) == 0
}
使用一个队列,push之后将n-1个元素依次放入队列尾部,pop时候,直接取出即可。
type MyStack struct {
q []int
}
func Constructor() MyStack {
return MyStack{}
}
func (s *MyStack) Push(x int) {
s.q = append(s.q, x)
for n := len(s.q); n > 1; n-- {
s.q = append(s.q, s.q[0])
s.q = s.q[1:]
}
}
func (s *MyStack) Pop() int {
v := s.q[0]
s.q = s.q[1:len(s.q)]
return v
}
func (s *MyStack) Top() int {
return s.q[0]
}
func (s *MyStack) Empty() bool {
return len(s.q) == 0
}