文章:代码随想录
状态:对stack,queue的实例化和方法不熟悉
leecode 232 用栈实现队列
思路:
用两个栈实现。核心思想就是将前一个栈的元素全部压入第二个栈中,这样就实现了元素倒序。然后再执行pop/peek。
class MyQueue {
Stack<Integer> inStack;
Stack<Integer> outStack;
public MyQueue() {
inStack=new Stack<Integer>();
outStack=new Stack<Integer>();
}
public void push(int x) {
inStack.push(x);
}
public int pop() {
//如果outStack中还有元素的话,那么就证明之前的元素还没有pop完,所以不能将新加入的元素直接压入第二个栈中,不然顺序就乱了。
if(!outStack.empty()){return outStack.pop();}
inOut();
//第二种写法
// if(outStack.empty()){inOut();}
return outStack.pop();
}
public int peek() {
if(!outStack.empty()){return outStack.peek();}
inOut();
//第二种写法
// if(outStack.empty()){inOut();}
return outStack.peek();
}
public boolean empty() {
return inStack.empty() && outStack.empty();
}
public void inOut(){
while(!inStack.empty()){
outStack.push(inStack.pop());
}
}
}
Leecode 225 用队列实现栈
思路:
核心思想就是在每个元素加入的时候就让他们以倒序的顺序在队列中存在. 因为队列有两端开口,可以在push处这么实现(stack就不行--上一题),更容易实现也少了两次边界条件的判断(pop+peek). 将最后一位之前的全部元素pop出来再重新加入队列的尾部。进行size-1次pop,下一个就是需要放在头部的元素(倒序)。
要注意Queue是接口,有很多实现这个接口的类
注意不同类的API的调用.
代码:
class MyStack {
Queue<Integer> que;
public MyStack() {
que=new ArrayDeque<>();
}
public void push(int x) {
que.offer(x);
//将之前进入的元素放在后面实现倒序
for (int i = 0; i < que.size()-1; i++) {
que.offer(que.poll());
}
}
public int pop() {
return que.poll();
}
public int top() {
return que.peek();
}
public boolean empty() {
return que.isEmpty();
}
}