题目:用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路:
我们要保证出栈的栈顶元素是先被push的元素,而一般出栈的栈顶元素是后被push的元素。所以,我们需要调整栈中元素的次序。不妨设我们总是用第二个栈s2出栈的元素作为出队元素,那么栈s2中元素的次序必定是调整过的。这个调整过程可以使用栈s1来完成。
解法1:初始状态下,栈s2都为空。将入队元素push进s1,若需要出栈,将s1中元素依次push入s2,再将s2中的栈中元素出栈。出栈完毕后,再将s2中元素依次push入s1,这样又回到了初始状态。循环这个过程就得到了一个队列。
对应的代码如下:
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);
return;
}
public int pop() {
while(!stack1.empty()){
stack2.push(stack1.pop());
}
int result = stack2.pop();
while(!stack2.empty()){
stack1.push(stack2.pop());
}
return result;
}
}
解法2:解法一存在缺点:s2中元素被push入s1后,当下次出栈时,这部分元素仍要被push入s2,显然这是冗余的操作,降低了算法的性能。所以考虑进入s2中的元素不再移动。所以当s2中元素全部出栈前(即s2不空时),不能再向s2中push元素。
由此思路得到解法2,。对应的代码如下:
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);
return;
}
public int pop() {
if(!stack2.empty()){
return stack2.pop();
}else{
while(!stack1.empty()){
stack2.push(stack1.pop());
}
return stack2.pop();
}
}
}
体会一下这句话:复杂系统总是来源于简单系统的演化。