用栈实现队列
题目:使用栈实现队列的下列操作:
push(x) -- 将一个元素放入队列的尾部。 pop() -- 从队列首部移除元素。 peek() -- 返回队列首部的元素。 empty() -- 返回队列是否为空。
示例:
MyQueue queue = new MyQueue(); queue.push(1); queue.push(2); queue.peek(); // 返回 1 queue.pop(); // 返回 1 queue.empty(); // 返回 false
说明:
-
你只能使用标准的栈操作 -- 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
-
你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
-
假设所有操作都是有效的 (例如,一个空的队列不会调用 pop 或者 peek 操作)。
力扣232题
思路
-
这是一道模拟题,不涉及到具体算法,考察的就是对栈和队列的掌握程度。
-
使用栈来模式队列的行为,如果仅仅用一个栈,是一定不行的,所以需要两个栈一个输入栈,一个输出栈,这里要注意输入栈和输出栈的关系。
下面动画模拟以下队列的执行过程:
执行语句: queue.push(1); queue.push(2); queue.pop(); 注意此时的输出栈的操作 queue.push(3); queue.push(4); queue.pop(); queue.pop();注意此时的输出栈的操作 queue.pop(); queue.empty();
-
在push数据的时候,只要数据放进输入栈就好,但在pop的时候,操作就复杂一些,输出栈如果为空,就把进栈数据全部导入进来(注意是全部导入),再从出栈弹出数据,如果输出栈不为空,则直接从出栈弹出数据就可以了。
-
最后如何判断队列为空呢?如果进栈和出栈都为空的话,说明模拟的队列为空了。
在代码实现的时候,会发现pop() 和 peek()两个函数功能类似,代码实现上也是类似的
class MyQueue { public: stack<int> stackIn; //利用两个栈来模拟队列,stackin进行进队列操作 stack<int> stackOut; //将进队列的元素加到stackout中,然后进行出栈,保持队列的先进先出 MyQueue() { } void push(int x) { stackIn.push(x); //进队列直接放入stackin中 } int pop() { // 只有当stOut为空的时候,再从stIn里导入数据(导入stIn全部数据) if(stackOut.empty()){ //因为要保证队列的先进先出,所以先判断stackout是否为空 // 从stIn导入数据直到stIn为空 while(!stackIn.empty()){ //将stackin的元素加入stackout,保证出队列的顺序 stackOut.push(stackIn.top()); //将stackin栈顶元素加入stackout stackIn.pop(); //将栈顶出栈 } } int result= stackOut.top(); //完成后stackout栈顶元素就是出队列的第一个元素 stackOut.pop();//将stackout栈顶元素出栈 return result; } int peek() { //返回队列开头的元素 int res=this->pop();//直接调用pop函数(因为peek函数中大部分也是复用了pop中代码) stackOut.push(res); //this->pop取得队列中开头元素(stackout栈顶元素)后将其出栈了,题目只是返回,所以还得压入栈中 return res; } bool empty() { return stackIn.empty()&&stackOut.empty(); //如果stackin和stackout都没元素则队列为空 } }; /** * 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(); * bool param_4 = obj->empty(); */
-
时间复杂度: push和empty为O(1), pop和peek为O(n)
-
空间复杂度: O(n)
注意:一定要懂得复用,功能相近的函数要抽象出来,不要大量的复制粘贴,很容易出问题!
用队列实现栈
使用队列实现栈的下列操作:
-
push(x) -- 元素 x 入栈
-
pop() -- 移除栈顶元素
-
top() -- 获取栈顶元素
-
empty() -- 返回栈是否为空
注意:
-
你只能使用队列的基本操作-- 也就是 push to back, peek/pop from front, size, 和 is empty 这些操作是合法的。
-
你所使用的语言也许不支持队列。 你可以使用 list 或者 deque(双端队列)来模拟一个队列 , 只要是标准的队列操作即可。
-
你可以假设所有操作都是有效的(例如, 对一个空的栈不会调用 pop 或者 top 操作)。
力扣225题
思路
-
依然想着用一个输入队列,一个输出队列,就可以模拟栈的功能,仔细想一下还真不行!
队列模拟栈,其实一个队列就够了,那么我们先说一说两个队列来实现栈的思路。
队列是先进先出的规则,把一个队列中的数据导入另一个队列中,数据的顺序并没有变,并没有变成先进后出的顺序。
-
所以用栈实现队列, 和用队列实现栈的思路还是不一样的,这取决于这两个数据结构的性质。
但是依然还是要用两个队列来模拟栈,只不过没有输入和输出的关系,而是另一个队列完全用来备份的!
如下面动画所示,用两个队列que1和que2实现队列的功能,que2其实完全就是一个备份的作用,把que1最后面的元素以外的元素都备份到que2,然后弹出最后面的元素,再把其他元素从que2导回que1。
模拟的队列执行语句如下:
queue.push(1); queue.push(2); queue.pop(); // 注意弹出的操作 queue.push(3); queue.push(4); queue.pop(); // 注意弹出的操作 queue.pop(); queue.pop(); queue.empty();
利用两个队列实现
class MyStack { public: queue<int> que1; queue<int> que2; // 辅助队列,用来备份 MyStack() { } void push(int x) { que1.push(x); //直接将元素存入栈里(也就是队列que1中) } int pop() { int result=0; if(que2.empty()){ while(!que1.empty()){ //将que1中的元素全部备份到que2中,除了que1最后一个元素,也就是栈顶元素,因为后面需要出栈 if(que1.size()==1){ //如果长度que1等于1,说明只剩栈顶元素,跳出循环 ,而且该判断一定要写在前面,防止只有一个元素时进入循环直接被出队列,从而后面result得到异常值, 当然不建议这么写,que1.size()底层会利用循环计算,从而增加时耗 break; } que2.push(que1.front());//将que1队列中队头元素放到队列2中 que1.pop(); //出队 } result=que1.front();//记录que1的最后一个元素,也对应了栈顶元素 que1.pop(); //出队列 也就是出栈 while(!que2.empty()){ //再将que2中的元素放入到que1中,此时已经不包含出队的那个元素 que1.push(que2.front()); que2.pop(); } } return result; //返回 } int top() { return que1.back(); //直接返回que1队列的队尾元素,也对应栈顶元素 } bool empty() { return que1.empty();//只需判断que1是否为空,因为que2是用来备份的 } }; /** * 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(); * bool param_4 = obj->empty(); */
优化
class MyStack { public: queue<int> que1; queue<int> que2; // 辅助队列,用来备份 MyStack() { } void push(int x) { que1.push(x); //直接将元素存入栈里(也就是队列que1中) } int pop() { int size=que1.size(); size--;//因为需要只剩下栈顶元素,从而循环 que1.size() -1 while(size--){ //将que1中的元素全部备份到que2中,除了que1最后一个元素,也就是栈顶元素,因为后面需要出栈 que2.push(que1.front());//将que1队列中队头元素放到队列2中 que1.pop(); //出队 } int result=que1.front();//记录que1的最后一个元素,也对应了栈顶元素 que1.pop(); //出队列 也就是出栈 que1=que2; // 再将que2赋值给que1 while(!que2.empty()){// 清空que2 que2.pop(); } return result; //返回 } int top() { return que1.back(); //直接返回que1队列的队尾元素,也对应栈顶元素 } bool empty() { return que1.empty();//只需判断que1是否为空,因为que2是用来备份的 } }; /** * 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(); * bool param_4 = obj->empty(); */
-
时间复杂度: pop为O(n),其他为O(1)
-
空间复杂度: O(n)
利用一个队列实现
其实这道题目就是用一个队列就够了。
一个队列在模拟栈弹出元素的时候只要将队列头部的元素(除了最后一个元素外) 重新添加到队列尾部,此时再去弹出元素就是栈的顺序了。
class MyStack { public: queue<int> que1; MyStack() { } void push(int x) { que1.push(x); //直接将元素存入栈里(也就是队列que1中) } int pop() { int size=que1.size(); size--;//因为需要只剩下原栈顶元素,从而循环que1.size() -1 while(size--){ //将que1中的元素全部再次添加到que1尾部,除了原本que1最后一个元素,也就是栈顶元素,因为后面需要出栈 // 将队列头部的元素(除了最后一个元素外) 重新添加到队列尾部 que1.push(que1.front());//将que1队列中队头元素放到队列2中 que1.pop(); //出队 } int result=que1.front();//记录que1的原队列最后一个元素,也对应了栈顶元素 // 此时弹出的元素顺序就是栈的顺序了 que1.pop(); //出队列 也就是出栈 return result; //返回 } int top() { return que1.back(); //直接返回que1队列的队尾元素,也对应栈顶元素 } bool empty() { return que1.empty();//只需判断que1是否为空,因为que2是用来备份的 } }; /** * 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(); * bool param_4 = obj->empty(); */
-
时间复杂度: pop为O(n),其他为O(1)
-
空间复杂度: O(n)