LeetCode a Day（六）——剑指offer 09.用两个栈实现队列

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

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

示例1：

输入：
[“CQueue”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[3],[],[]]
输出：
[null,null,3,-1]

示例2：

输入：
[“CQueue”,“deleteHead”,“appendTail”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[],[5],[2],[],[]]
输出：
[null,-1,null,null,5,2]

考虑到这两个示例不太好懂，这里稍稍解释一下。
输入第一行的字符串是指调用的函数，CQueue是构造函数，deleteHead指删除头元素，appendTail指添加尾元素。
第二行指各函数对应的参数。
输出则是各函数的返回值。

思路与解法（一）——一个栈用于添加，一个栈用于删除

我们知道栈是先入后出FILO的，若仅有一个栈，想直接取出队列的头元素是不可能的，因为它在栈底。
那么两个栈怎么做呢？这里基于的原理是，将Stack1中的元素依次先pop再push进Stack2，可以实现倒序。
原本在栈底的元素变为栈顶。

这样Stack1做添加尾元素，Stack2做删除头元素即可。

Java实现如下：

class CQueue {
    Stack<Integer> s1, s2;
    public CQueue() {
        s1 = new Stack<>();
        s2 = new Stack<>();
    }
    
    public void appendTail(int value) {
        s1.push(value);
    }
    
    public int deleteHead() {
        if(s2.isEmpty()){
            if(s1.isEmpty()){
                return -1;
            }
            while(!s1.isEmpty()){
                s2.push(s1.pop());
            }
        }
        return s2.pop();
    }
}

复杂度分析

时间复杂度：appendTail函数时间复杂度为O(1)。deleteHead删除n次头元素需要对n个元素实现倒序。
空间复杂度：O(n)，最坏情况下两个栈保存n个元素。