题目:225. 用队列实现栈
请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push
、top
、pop
和 empty
)。
实现 MyStack
类:
void push(int x)
将元素 x 压入栈顶。int pop()
移除并返回栈顶元素。int top()
返回栈顶元素。boolean empty()
如果栈是空的,返回true
;否则,返回false
。
思路:
让两个先进先出的队列分工明确,保证每次进行插入删除操作时,有一个队列为空,空的队列来进行操作
public class MyStack {
// 实现一个先进后出的栈
// 队列 先进先出
Queue<Integer> q1 ;
Queue<Integer> q2 ;
public MyStack() {
q1 = new LinkedBlockingQueue<>();
q2 = new LinkedBlockingQueue<>();
}
public void push(int x) {
if(q1.size() == 0){
q1.add(x);
while(q2.size() != 0){
q1.add(q2.poll());
}
}else if(q2.size() == 0){
q2.add(x);
while(q1.size() != 0){
q2.add(q1.poll());
}
}
}
public int pop() {
if(q1.size() == 0){
return q2.poll();
}else if(q2.size() == 0){
return q1.poll();
}
return -1;
}
public int top() {
if(q1.size() == 0){
return q2.peek();
}else if(q2.size() == 0){
return q1.peek();
}
return -1;
}
public boolean empty() {
return q1.size() == 0 && q2.size() == 0;
}
}
/**
* 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();
*/
题目:232. 用栈实现队列
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push
、pop
、peek
、empty
):
实现 MyQueue
类:
void push(int x)
将元素 x 推到队列的末尾int pop()
从队列的开头移除并返回元素int peek()
返回队列开头的元素boolean empty()
如果队列为空,返回true
;否则,返回false
思路:
一个栈s1专门用作入栈,另一个栈s2专门用作出栈
class MyQueue {
// 入栈
Stack<Integer> s1;
// 出栈
Stack<Integer> s2;
public MyQueue() {
s1 = new Stack<>();
s2 = new Stack<>();
}
public void push(int x) {
s1.push(x);
}
public int pop() {
if (s2.isEmpty()) {
while (s1.size() != 0) {
s2.push(s1.pop());
}
}
return s2.pop();
}
public int peek() {
if (s2.isEmpty()) {
while (s1.size() != 0) {
s2.push(s1.pop());
}
}
return s2.peek();
}
public boolean empty() {
return s1.size() == 0 && s2.size() == 0;
}
}
/**
* 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();
*/