剑指 Offer 04. 用两个栈实现队列
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路:一个栈只负责压入元素,另一个栈在不为空的情况下弹栈,为空的时候先将Stack1全部压栈,再弹出结果。
package main var stack1 [] int var stack2 [] int func Push(node int) { stack1 = append(stack1, node) } func Pop() int{ res := -1 if len(stack2) > 0 { res = stack2[0] stack2 = stack2[1:] } else { if len(stack1) != 0 { res = stack1[0] stack2 = stack1[1:] stack1 = stack1[:0] } } return res }