用队列实现栈
题目描述:请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通队列的全部四种操作(push、top、pop 和 empty)。
实现 MyStack 类:
void push(int x) 将元素 x 压入栈顶。
int pop() 移除并返回栈顶元素。
int top() 返回栈顶元素。
boolean empty() 如果栈是空的,返回 true ;否则,返回 false 。
示例说明请见LeetCode官网。
解法一:双队列实现栈
使用2个队列firstQueue和secondQueue存储数据,具体方法说明如下:
push(int x):如果firstQueue为空,则将x存到secondQueue中,否则存到firstQueue中;
pop():如果firstQueue为空,则将secondQueue中的数据都取出然后依次存入firstQueue中只留一个作为栈顶元素取出并返回;否则,将firstQueue中的数据都取出然后依次存入secondQueue中只留一个作为栈顶元素取出并返回;
top():逻辑通pop()方法类似;
empty():如果firstQueue和secondQueue都为空,返回true;否则,返回false。
import java.util.LinkedList;
import java.util.Queue;
public class LeetCode_225 {
public static void main(String[] args) {
MyStack myStack = new MyStack();
myStack.push(1);
myStack.push(2);
myStack.push(3);
System.out.println(myStack.top()); // 返回 2
System.out.println(myStack.pop()); // 返回 2
System.out.println(myStack.empty()); // 返回 False
}
}
class MyStack {
private Queue<Integer> firstQueue;
private Queue<Integer> secondQueue;
/**
* Initialize your data structure here.
*/
public MyStack() {
firstQueue = new LinkedList<>();
secondQueue = new LinkedList<>();
}
/**
* Push element x onto stack.
*/
public void push(int x) {
if (firstQueue == null) {
secondQueue.add(x);
} else {
firstQueue.add(x);
}
}
/**
* Removes the element on top of the stack and returns that element.
*/
public int pop() {
if (this.empty()) {
throw new RuntimeException("empty stack.");
}
if (firstQueue.isEmpty()) {
int size = secondQueue.size();
for (int i = 1; i < size; i++) {
firstQueue.add(secondQueue.poll());
}
return secondQueue.poll();
} else {
int size = firstQueue.size();
for (int i = 1; i < size; i++) {
secondQueue.add(firstQueue.poll());
}
return firstQueue.poll();
}
}
/**
* Get the top element.
*/
public int top() {
if (this.empty()) {
throw new RuntimeException("empty stack.");
}
int result = -1;
if (firstQueue.isEmpty()) {
int size = secondQueue.size();
for (int i = 1; i < size; i++) {
firstQueue.add(secondQueue.poll());
}
result = secondQueue.peek();
firstQueue.add(secondQueue.poll());
} else {
int size = firstQueue.size();
for (int i = 1; i < size; i++) {
secondQueue.add(firstQueue.poll());
}
result = firstQueue.peek();
secondQueue.add(firstQueue.poll());
}
return result;
}
/**
* Returns whether the stack is empty.
*/
public boolean empty() {
return firstQueue.isEmpty() && secondQueue.isEmpty();
}
}