代码随想录算法训练营第9天 | 栈和队列部分:232. 用栈实现队列, 225. 用队列实现栈
栈和队列部分
基础知识:
栈:先进后出
队列:先进先出
栈相关类:Stack(实现类),不推荐使用,因为继承Vector,Vector线程安全,所以Stack也是线程安全的,但是消耗资源较大,所以不推荐使用
队列相关类:Queue(单向队列接口), Deque (双向队列接口,同时具备栈和队列的功能, 常用实现类包括ArrayDeque,LinkedList,推荐使用ArrayDeque, 双向队列的话可以实现双端的插入和删除)
232. 用栈实现队列
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push、pop、peek、empty):
实现 MyQueue 类:
void push(int x) 将元素 x 推到队列的末尾
int pop() 从队列的开头移除并返回元素
int peek() 返回队列开头的元素
boolean empty() 如果队列为空,返回 true ;否则,返回 false
说明:你只能使用标准的栈操作 —— 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
思路:采用两个栈来实现队列的功能
一个输入栈,一个输出栈
经过两次栈的操作之后就会回到原来的顺序
但是需要注意的是:如果输出栈里面还有元素的话,是不需要将输入栈里面的元素入到输出栈里面的,因为这样的话会混乱原有的顺序,使得后一波的输入的第一个元素成为了队列中的下一个元素,而不是输出栈里面的剩下的元素
如果要是输出栈为空的话,再往从输入栈往输出栈里面补充元素
补充:类中输入,输出栈声明就可以,在构造器里面初始化
class MyQueue {
public Stack<Integer> sin=new Stack<>();//构造两个栈,一个输入栈,一个输出栈
public Stack<Integer> sout=new Stack<>();
public MyQueue() {
}
public void push(int x) {
sin.push(x);
}
public int pop() {
if(!sout.isEmpty()){
return sout.pop();//输出栈不空,直接输出
}else{
while(!sin.isEmpty()){
int x=sin.pop();
sout.push(x);//输入栈空的话,将输入栈里面所有的元素向输出栈入栈
}
return sout.pop();
}
}
public int peek() {
if(!sout.isEmpty()){
return sout.peek();//这边同理
}else{
while(!sin.isEmpty()){
int x=sin.pop();
sout.push(x);
}
return sout.peek();
}
}
public boolean empty() {
return sin.isEmpty() && sout.isEmpty();
}
}
225. 用队列实现栈
请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push、top、pop 和 empty)。
实现 MyStack 类:
void push(int x) 将元素 x 压入栈顶。
int pop() 移除并返回栈顶元素。
int top() 返回栈顶元素。
boolean empty() 如果栈是空的,返回 true ;否则,返回 false 。
注意:你只能使用队列的基本操作 —— 也就是 push to back、peek/pop from front、size 和 is empty 这些操作。你所使用的语言也许不支持队列。 你可以使用 list (列表)或者 deque(双端队列)来模拟一个队列 , 只要是标准的队列操作即可。
思路:使用两个队列来实现栈的功能,先把除了最后一个元素之外的所有元素移动到辅助队列中,然后再从辅助队列中移动回来。
需要注意的点是:queue2作为辅助队列的时候,把内容还给queue1的时候,需要把自己清空,我这边是用循环把里面的元素全都退给了queue1,看看答案有没有更好的方式
另外一方面:存在size()函数可以表示队列的大小,不然这个题我真的不知道怎么做
class MyStack {
Deque<Integer> queue1;//主队列
Deque<Integer> queue2;//辅助队列
public MyStack() {
queue1=new LinkedList<>();
queue2=new LinkedList<>();
}
public void push(int x) {
queue1.add(x);//这边是正常的
}
public int pop() {
while(queue1.size()>1){
queue2.add(queue1.poll());
}
int res=queue1.poll();
//queue1=queue2;//这边出现错误,queue1指向queue2,queue2并没有发生变化,之后再操作的话就会产生错误
//queue2作为辅助队列,用完之后就应该被清空
while(!queue2.isEmpty()){
queue1.add(queue2.poll());//没错,就是这边的问题,改了之后就可以了
}
return res;
}
public int top() {
while(queue1.size()>1){
queue2.add(queue1.poll());
}
int res=queue1.peek();
queue2.add(queue1.poll());
while(!queue2.isEmpty()){
queue1.add(queue2.poll());
}
return res;
}
public boolean empty() {
return queue1.isEmpty();
}
}
总结:
栈的实现类:Stack: push(int x), pop(), isEmpty(), top()
Deque接口的实现:LinkedList:offer(int x), add(int x), remove(),poll(), peek(), isEmpty(),size();