题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
就……字面意思,进栈的时候用一个栈来存数据,另一个栈通过中间变量将数据倒着存一遍
出栈的时候将倒着存的那个栈栈顶pop,然后流程和进栈差不多(……)
需要注意的是每次倒腾都要把被存放的栈清空啊,因为没注意这个wa了两次(。)
import java.util.Stack;
public class Solution {
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
public void push(int node) {
stack1.push(node);
Stack<Integer> tmp = (Stack<Integer>) stack1.clone();
stack2.clear();
while(!tmp.empty()) {
stack2.push(tmp.pop());
}
}
public int pop() {
int k = stack2.pop();
Stack<Integer> tmp = (Stack<Integer>) stack2.clone();
stack1.clear();
while(!tmp.empty()) {
stack1.push(tmp.pop());
}
return k;
}
}
然后看题解里的大佬根本没用中间变量倒腾(。)所以码一下题解
https://blog.nowcoder.net/n/ee03f62d5a6e4b5ea571d61ad7eb64c8?f=comment
队列的特性是:“先入先出”,栈的特性是:“先入后出”
当我们向模拟的队列插入数 a,b,c 时,假设插入的是 stack1,此时的栈情况为:
栈 stack1:{a,b,c} 栈 stack2:{}
当需要弹出一个数,根据队列的"先进先出"原则,a 先进入,则 a 应该先弹出。但是此时 a 在 stack1 的最下面,将 stack1
中全部元素逐个弹出压入 stack2,现在可以正确的从 stack2 中弹出 a,此时的栈情况为:栈 stack1:{} 栈 stack2:{c,b}
继续弹出一个数,b 比 c 先进入"队列",b 弹出,注意此时 b 在 stack2 的栈顶,可直接弹出,此时的栈情况为:
栈 stack1:{} 栈 stack2:{c}
此时向模拟队列插入一个数 d,还是插入 stack1,此时的栈情况为:
栈 stack1:{d} 栈 stack2:{c}
弹出一个数,c 比 d 先进入,c 弹出,注意此时 c 在 stack2 的栈顶,可直接弹出,此时的栈情况为:
栈 stack1:{d} 栈 stack2:{c}
根据上述栗子可得出结论:
当插入时,直接插入 stack1 当弹出时,当 stack2 不为空,弹出 stack2 栈顶元素,如果 stack2 为空,将 stack1 中的全部数逐个出栈入栈 stack2,再弹出 stack2 栈顶元素
链接:https://www.nowcoder.com/questionTerminal/54275ddae22f475981afa2244dd448c6?answerType=1&f=discussion
来源:牛客网
import java.util.Stack;
public class Solution {
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
public void push(int node) {
stack1.push(node);
}
public int pop() {
if (stack2.size() <= 0) {
while (stack1.size() != 0) {
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
}