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

news2024/9/25 15:24:06

题目

在这里插入图片描述

解题

这道题是用两个栈实现一个队列，具有一个队尾添加元素和队头删除元素的功能
这里思路是一个进，一个出，里面的元素顺序很关键，要保持一个进栈最先进的元素在另一个出栈的栈顶
第一种方法是，每次进元素，就把另一个栈的元素重排，放到最下面
第二种方法是,每次进元素，就初始化另一个栈，然后这个栈元素依次进另一个栈
第一种明显太复杂，使用第二种
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();
    }

}