剑指offer
题目要求
用两个栈来实现一个队列,完成队列的Push(入栈)和Pop(出栈)操作。
队列中的元素为int类型
补充
栈的特性:先进后出(LIFO, Last In First Out),类似于水桶。
队列的特性:先进先出(FIFO, FIFO In First Out),类似于水管。
核心思想
利用两个栈,一个用于入队,一个用于出队,入队时,直接放入栈1,出队时,由栈1倒到栈2,再出(将先出栈的又变成先入栈的,后出栈的变成后入栈的,因此达到队列的要求,通俗的讲,就是负负得正)
完整代码如下
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() {
//如果栈2是空,就把栈1内容弹出来入栈2,如果上述操作后仍然为空,则返回-1;
if(stack2.empty()){
//这里,只有在stack2为空时,即pop完之后,才可以插入stack1的元素。否则会发生错误
while(stack1.empty() != true){
stack2.push(stack1.pop());
}
}
if(stack2.empty()){
//即stack1也为空
return -1;
}
//否则,可以出队(出栈)
return stack2.pop();
}
public static void main(String[] args) {
Solution sl = new Solution();
sl.push(6);
sl.push(7);
sl.push(8);
System.out.println(sl.pop());
System.out.println(sl.pop());
}
}