题目1:用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
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(stack1.empty() && stack2.empty()){
throw new RuntimeException("Queue is empty!");
}
if(stack2.empty()){
while(!stack1.empty()){
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
}
题目2,用两个队列实现栈
public class Solution {
Queue<integer> queue1 = new ArrayDeque<>();
Queue<integer> queue2 = new ArrayDeque<>();
public void push(int node) {
//两个栈都为空时,优先考虑queue1
if (queue1.isEmpty()&&queue2.isEmpty()) {
queue1.add(node);
return;
}
//如果queue1为空,queue2有元素,直接放入queue2
if (queue1.isEmpty()) {
queue2.add(node);
return;
}
//如果queue2为空,queue1有元素,直接放入queue1
if (queue2.isEmpty()) {
queue1.add(node);
return;
}
}
public int pop() {
//两个栈都为空时,没有元素可以弹出
if (queue1.isEmpty()&&queue2.isEmpty()) {
try {
throw new Exception(stack is empty);
} catch (Exception e) {
}
}
//如果queue1为空,queue2有元素, 将queue2的元素依次放入queue1中,直到最后一个元素,我们弹出。
if (queue1.isEmpty()) {
while (queue2.size()>1) {
queue1.add(queue2.poll());
}
return queue2.poll();
}
if (queue2.isEmpty()) {
while (queue1.size()>1) {
queue2.add(queue1.poll());
}
return queue1.poll();
}
return (Integer) null;
}
}