代码随想录第10天 | 第五章 栈与队列part01

代码随想录算法训练营第10天 | 第五章 栈与队列part01 232.用栈实现队列 225. 用队列实现栈

题目一 232.用栈实现队列

使用栈实现队列的下列操作：
push(x) – 将一个元素放入队列的尾部。
pop() – 从队列首部移除元素。
peek() – 返回队列首部的元素。
empty() – 返回队列是否为空。

本题是用栈去实现队列的功能。题目也明确了方法：使用两个栈实现此功能，一个输入栈一个输出栈。两次先进后出便是先进先出。
本题关键反倒不是思考的过程，而是对各语言自己封装的栈一个熟悉和运用的过程，无论是c++ STL的Stack，还是java自己的Stack类和Deque类。毕竟不用真的自己造轮子。
push的时候直接让输入栈push即可；而输入栈挪到输出栈的过程在pop()中解决。
输出栈为空的时候，就需要将输入栈所有的元素依次吸过来。
然后挨个扔出去即可。
注意peek()函数只获取栈顶却不抛；pop()函数既返回栈顶还会抛掉；
唯一值得注意的是STL中Stack喜欢用top()获取栈顶元素，而Stack类喜欢用peek().

import java.util.*;
/*
* Stack in Java:
* push() pop() empty() peek() search()
*/
class MyQueue {
    private Stack<Integer> sin = new Stack<Integer>();
    private Stack<Integer> sout = new Stack<Integer>();

    public MyQueue() {
    }
    
    public void push(int x) {
        sin.push(x);
    }
    
    public int pop() {
        if(sout.empty() == true)
        {
            while(sin.empty() == false)
            {
                sout.push( sin.peek() );
                sin.pop();
            }
        }

        int result = sout.peek();
        sout.pop();
        return result;
    }

    public int peek() {
        int res = this.pop();
        sout.push(res);
        return res;
    }

    public boolean empty() {
        return sin.empty() && sout.empty();
    }
}

题目二 225. 用队列实现栈

使用队列实现栈的下列操作：

• push(x) – 元素 x 入栈
• pop() – 移除栈顶元素
• top() – 获取栈顶元素
• empty() – 返回栈是否为空

这题如果使用双向队列则毫无难度，因为deque既是队列又是栈。
那如果只用单向队列如何实现栈呢？

如果使用一个队列实现栈，那么就让元素挨个弹出再依次加入到队列里循环，只不过目标的那个元素弹出后就不回来了。
因此需要对元素进行标记和判断。计算好size()之后每次弹出对size进行自减，把每次弹出的元素都加回队列来。加回来结束之后，弹出并记录应当弹出的元素。

class MyStack {
    Queue<Integer> queue;
    
    public MyStack() {
        queue = new LinkedList<>();
    }
    
    public void push(int x) {
        queue.add(x);
    }
    
    public int pop() {
        rePosition();
        return queue.poll();
    }
    
    public int top() {
        rePosition();
        int result = queue.poll();
        queue.add(result);
        return result;
    }
    
    public boolean empty() {
        return queue.isEmpty();
    }

    public void rePosition()//弹出又加回来
    {
        int size = queue.size();
        size--;
        while(size-->0)
            queue.add(queue.poll());
    }
}

本题使用两个队列来实现栈。但是由于队列的特性，像上面那样用输出/输栈来实现栈完全不可能。
这里第二个队列只起到一个额外的存储空间，起到备份作用。
每次移动时先让全部元素入第一个栈，之后将除了最后一个元素外的，前面的其他元素全部挪到第二个队列处，给最后一个元素腾位(因为它应当第一个出去)，然后出去后再将其他元素原封不动的再挪回第一个队列。接着重复上述过程。

class MyStack {
    //q1作为主要的队列，其元素排列顺序和出栈顺序相同
    Queue<Integer> q1 = new ArrayDeque<>();
    //q2仅作为临时放置
    Queue<Integer> q2 = new ArrayDeque<>();

    public MyStack() {

    }
    //在加入元素时先将q1中的元素依次出栈压入q2，然后将新加入的元素压入q1，再将q2中的元素依次出栈压入q1
    public void push(int x) {
        while (q1.size() > 0) 
        {
            q2.add(q1.poll());//挪过来
        }
        q1.add(x);
        while (q2.size() > 0) 
        {
            q1.add(q2.poll());//挪回去
        }
    }

    public int pop() {
        return q1.poll();
    }

    public int top() {
        return q1.peek();
    }

    public boolean empty() {
        return q1.isEmpty();
    }
}