剑指 Offer 04. 用两个栈实现队列

剑指 Offer 04. 用两个栈实现队列

用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。

思路：一个栈只负责压入元素，另一个栈在不为空的情况下弹栈，为空的时候先将Stack1全部压栈，再弹出结果。

 

package main

var stack1 []int // 3,4
var stack2 []int // 1,2

func Push(node int) {
	stack1 = append(stack1, node)
}

func Pop() int {
    if len(stack2) == 0 && len(stack1) == 0 {
        return -1
    }
    
	if len(stack2) == 0 {
		stack2 = stack1
        stack1 = make([]int, 0)
	}

	tmp := stack2[0]
	stack2 = stack2[1:]
	return tmp
}