两个普通栈实现队列
题目
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push、pop、peek、empty):
实现 MyQueue 类:
void push(int x): 将元素 x 推到队列的末尾
int pop() :从队列的开头移除并返回元素
int peek(): 返回队列开头的元素
boolean empty(): 如果队列为空,返回 true ;否则,返回 false
栈: 先进后出
队列:先进先出
两个栈实现队列
个人思路:俩个栈 s1 s2
s1只用来入栈
s2只用来出栈
出栈的时候 如果s2为空,就判断 s1是否也为空
如果s1也为空 就 return
否则 将s1的全部倒进s2 从s1 pop出来,s2 push进去
for 循环条件 不能 直接 i<s1.size(); 因为是一个变量 每次出一个 就会减一
peek : 全部倒进 s2 后 s2.peek() ; 就好
class MyQueue {
public Stack<Integer> s1;
public 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()){
if(s1.isEmpty()){
return -1;
}
int size = s1.size();
for(int i = 0;i<size;i++){
s2.push(s1.pop());
}
}
int ret = s2.pop();
return ret;
}
public int peek() {
if(s2.isEmpty()){
if(s1.isEmpty()){
return -1;
}
int size = s1.size();
for(int i = 0;i<size;i++){
s2.push(s1.pop());
}
}
int ret = s2.peek();
return ret;
}
public boolean empty() {
return s1.isEmpty()&&s2.isEmpty();
}
}