代码随想录训练营第十天 | 232.用栈实现队列 225. 用队列实现栈

232.用栈实现队列

利用两个栈s1,s2一个模仿队头，一个模仿队尾。

主要是在于栈模仿队列时的pop(),peek()函数中判断模仿队头中是否还有元素（由于队列"先进先出"特性若有元素需要首先pop() 或 peek() ）

代码如下：

class MyQueue {
    // 创建两个栈,  s1 用于模仿队尾 | s2 用于模仿队头
    Stack<Integer> s1,s2;

    public MyQueue() {
        s1 = new Stack<>();
        s2 = new Stack<>();
    }
    
    public void push(int x) {
        // 入队操作直接加入s1
        s1.push(x);
    }
    
    public int pop() {
        peek();
        return s2.pop();
    }
    
    public int peek() {
        int temp = 0;
        if(s2.isEmpty()){ // 由于队列特性（先进先出） 所以我们需要先检查s2中是否为空(不为空就要直接 peek 或 pop)
            while(!s1.isEmpty()){
                temp = s1.pop();
                s2.push(temp);
            }
        }
        return s2.peek();
    }
    
    public boolean empty() {
        return s1.isEmpty() && s2.isEmpty();
    }
}

225. 用队列实现栈

由于队列是先进先出的规则，把一个队列中的数据导入另一个队列中，数据的顺序并没有变，并没有变成先进后出的顺序。

所以实际上一个队列便可以进行模仿栈操作，而创建另一个队列为了临时存储数据并在操作完成后返回。

代码如下：

class MyStack {
    // 由于队列先进先出特性，将一个队列元素存入另一个队列并未改变元素进出顺序
    // 所以只能在一个队列中下功夫
    Queue<Integer> q1,q2; // 另一个为临时队列用于存储元素
    public MyStack() {
        q1 = new LinkedList<>();
        q2 = new LinkedList<>();
    }
    
    public void push(int x) {
        q1.offer(x);
    }
    
    public int pop() {
        while(q1.size() > 1){
            q2.offer(q1.poll());
        } 
        while(!q2.isEmpty()){
            q1.offer(q2.poll());
        }
        return q1.poll();
    }
    
    public int top() {
        while(q1.size() > 1){
            q2.offer(q1.poll());
        }
        
        int top = q1.peek(); // 由于top操作不能改变内部结构,创建一个临时值来进行存储top值
        q2.offer(q1.poll()); // 将内部结构还原

        while(!q2.isEmpty()){
            q1.offer(q2.poll());
        }
        return top;
    }
    
    public boolean empty() {
        return q1.isEmpty();
    }
}