剑指Offer | 用两个栈实现对列
题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路
输入:
[“PSH1”,“PSH2”,“PSH3”,“POP”,“POP”,“PSH4”,“POP”,“PSH5”,“POP”,“POP”]
输出:
1,2,3,4,5
用stack1存放push的所有值,如果需要出列,首先判断stack2里面有没有值,如果没有的话,把stack1中的所有值推入stack2,返回stack2的栈顶
代码
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.isEmpty()) {
while(!stack1.isEmpty()) stack2.push(stack1.pop());
}
return stack2.pop();
}
}