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

题目

LeetCode 剑指 Offer 09. 用两个栈实现队列_第1张图片

解题

这道题是用两个栈实现一个队列,具有一个队尾添加元素和队头删除元素的功能
这里思路是一个进,一个出,里面的元素顺序很关键,要保持一个进栈最先进的元素在另一个出栈的栈顶
第一种方法是,每次进元素,就把另一个栈的元素重排,放到最下面
第二种方法是,每次进元素,就初始化另一个栈,然后这个栈元素依次进另一个栈
第一种明显太复杂,使用第二种
s1,s2两个栈
首先初始化就是new就可以
入栈时要先放进s1,然后s2要分情况,为空时直接入栈,不为空要先置空s1,再把s1的元素顺序入栈s2;然而想到这里突然发现这样s1出栈后没了,怎么挪进s2,然后保持s1原样是个问题;还有就是删除队尾时,s1最上面删掉,s2栈底没法删,只能初始化后再把s1的元素再全入s2栈
首先可以先试试能不能直接全部入栈后保持原样,这里想了很久,只有一种办法
入队列操作时,如果s2不为空,就把元素都放入s1,之后对s1入栈
如果出队列操作时,s1,s2都为空,就返回-1,如果s2为空,s1非空,就把元素倒入s2,s2的栈顶是最先进栈的元素,删掉即可
最后发现
之前一直想的是两个栈都有一批元素,陷入了思维误区,实际上操作的是这些元素,来回倒就行了

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

}

LeetCode 剑指 Offer 09. 用两个栈实现队列_第2张图片
这里把入队列只留一句
s1.push(value);也是可以的
如果想要优化时间,可以参考这个解法,使用双向链表,因为stack还是vector接口下的,本质是个object数组,效率较慢
LeetCode 剑指 Offer 09. 用两个栈实现队列_第3张图片
但是这样就用空间多了
LeetCode 剑指 Offer 09. 用两个栈实现队列_第4张图片

你可能感兴趣的:(leetcode,算法,java)