本次题目
- 232 用栈实现队列
232 用栈实现队列
- 使用两个栈:一个输入栈,一个输出栈。
- push时直接将数据压入输入栈,当pop时需要进行判断,如果输出栈为空,则将输入栈中的所有数据按弹出顺序压入输出栈,然后pop;如果输入栈不为空则直接pop。
- 判断栈顶元素peek同上述pop前,当输入栈和输出栈都为空时队列为空。
- 注意:在类中定义两个栈的数据结构,在初始化函数中进行初始化。
class MyQueue {
//定义数据结构
Stack<Integer> stackIn;
Stack<Integer> stackOut;
public MyQueue() {
//使用两个栈:一个输入栈,一个输出栈
stackIn = new Stack<>();
stackOut = new Stack<>();
}
public void push(int x) {
//push时直接将数据压入输入栈
stackIn.push(x);
}
public int pop() {
//当pop时需要进行判断
//如果输出栈为空,则将输入栈中的所有数据按弹出顺序压入输出栈
if(stackOut.isEmpty()){
while(!stackIn.isEmpty()){
stackOut.push(stackIn.pop());
}
}
//然后pop,如果输入栈不为空则直接pop
return stackOut.pop();
}
public int peek() {
//判断栈顶元素同pop弹出前
int res = this.pop();
//pop后要push回输出栈
stackOut.push(res);
return res;
}
public boolean empty() {
//当输入栈和输出栈都为空时队列为空
return stackIn.isEmpty() && stackOut.isEmpty();
}
}
/**
* Your MyQueue object will be instantiated and called as such:
* MyQueue obj = new MyQueue();
* obj.push(x);
* int param_2 = obj.pop();
* int param_3 = obj.peek();
* boolean param_4 = obj.empty();
*/