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

用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 appendTail 和 deleteHead ，分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素，deleteHead 操作返回 -1 )

示例 1：

输入：

["CQueue","appendTail","deleteHead","deleteHead","deleteHead"]

[[],[3],[],[],[]]

输出：[null,null,3,-1,-1]

示例 2：

输入：

["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]

[[],[],[5],[2],[],[]]

输出：[null,-1,null,null,5,2]

提示：

1 <= values <= 10000

最多会对 appendTail、deleteHead 进行 10000 次调用

来源：力扣（LeetCode）

链接：https://leetcode.cn/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

type CQueue struct {
    inStack, outStack []int //准备两个栈，一个作为入栈，一个做出栈
}

func Constructor() CQueue {
    return CQueue{}
}

func (this *CQueue) AppendTail(value int)  {
    this.inStack = append(this.inStack, value) //入队操作
}

func (this *CQueue) DeleteHead() int {
    if len(this.outStack) == 0 {
        if len(this.inStack) == 0 { //两个栈都为空，就返回-1
            return -1
        }
        this.in2out() //（第一轮的时候出栈必是空的）出栈为空，入栈有元素则使用in2out函数将入栈的元素倒入出栈
    }
    value := this.outStack[len(this.outStack)-1] //出队的队首元素是出栈的栈尾元素
    this.outStack = this.outStack[:len(this.outStack)-1] //更新出栈的元素
    return value //返回出队的元素值
}

func (this *CQueue) in2out() {
    for len(this.inStack) > 0 {//倒栈前提是入栈不为空
        this.outStack = append(this.outStack, this.inStack[len(this.inStack)-1])//把入栈元素一个个倒进出栈
        this.inStack = this.inStack[:len(this.inStack)-1] //更新入栈
    }
}


/**
 * Your CQueue object will be instantiated and called as such:
 * obj := Constructor();
 * obj.AppendTail(value);
 * param_2 := obj.DeleteHead();
 */