栈的常用方法在题目中,队列LinkedList的常用方法为poll() , add() , remove();本题和232. 用栈实现队列是栈和队列的相互实现经典题目,考察对栈和队列的理解。(记住队列实现栈只需要在push方法上改变即可,其他不变。)
用队列实现栈,首先想到用两个队列,只需要改变push()方法,其他复用队列方法即可。一个主队列,一个辅助队列,先向辅助队列add新元素,然后判断主队列是否为空,若为空则直接将辅助队列变为主队列,若不为空,则将主队列的元素poll并add辅助队列中,然后将辅助队列变为主队列,这样此队列可直接实现为栈。核心代码如下:
public void push(int x) {
q2.add(x);
while (!q1.isEmpty()) {
q2.add(q1.poll());
}
Queue<Integer> temp = q1;
q1 = q2;
q2 = temp;
}
另一种方法是使用一个队列实现,直接在原队列上改即可:
public void push(int x) {
int n = queue.size();
queue.offer(x);
for (int i = 0; i < n; i++) {
queue.offer(queue.poll());
}
}
最终实现:
class MyStack {
Queue<Integer> q1;
Queue<Integer> q2;
public MyStack() {
q1 = new LinkedList<>();
q2 = new LinkedList<>();
}
public void push(int x) {
q2.add(x);
while (!q1.isEmpty()) {
q2.add(q1.poll());
}
Queue<Integer> temp = q1;
q1 = q2;
q2 = temp;
}
public int pop() {
return q1.poll();
}
public int top() {
return q1.peek();
}
public boolean empty() {
return q1.isEmpty() && q2.isEmpty();
}
}
/**
* Your MyStack object will be instantiated and called as such:
* MyStack obj = new MyStack();
* obj.push(x);
* int param_2 = obj.pop();
* int param_3 = obj.top();
* boolean param_4 = obj.empty();
*/