赞
踩
首先我们在做栈和队列之前一定要做好基础知识的储备,包括一些底层逻辑 代码随想录
这里默认基础知识都会了:
Leetcode 232 用栈实现队列
题目链接 232 用栈实现队列
push()是输入元素
top()是取出栈顶元素,不会删掉栈里边的元素
pop()是删除栈顶元素
直接上代码加注释:
- class MyQueue {
- public:
- stack<int> stIn;//定义两个栈来模拟队列
- stack<int> stOut;
- MyQueue() {
-
- }
-
- void push(int x) {
- stIn.push(x);//push()是将元素输入栈中
- }
-
- int pop() {
- if(stOut.empty()){//首先要保证输出栈为空,要不然会影响输出元素的顺序
- while(!stIn.empty()){//只要输入栈没有空,就不停的将元素输出给输出栈
- stOut.push(stIn.top());//将输入栈的栈顶元素输出给输出栈
- stIn.pop();//删除输入站的栈顶元素
- }
-
- }
- int result = stOut.top();//用result记录输出栈的顶部元素
- stOut.pop();//移除顶部元素
- return result;
- }
-
- int peek() {//peek是用来返回顶元素的值
- int res = this->pop();//直接调用pop函数(这里删除了顶元素)
- stOut.push(res);//把顶元素补上,做到不删除元素
- return res;
- }
-
- bool empty() {
- return stIn.empty()&&stOut.empty();//如果为空,就返回空
- }
- };
-
- /**
- * 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();
- */
Leetcode 225 用队列实现栈
题目链接 225 用队列实现栈
该题目有两个方法,第一个方法:用两个队列que1和que2实现队列的功能,que2其实完全就是一个备份的作用,把que1最后面的元素以外的元素都备份到que2,然后弹出最后面的元素,再把其他元素从que2导回que1。
代码:
- class MyStack {
- public:
- queue<int> que1;
- queue<int> que2; // 辅助队列,用来备份
-
- MyStack() {
-
- }
- void push(int x) {
- que1.push(x);
- }
- int pop() {
- int size = que1.size();
- size--;
- while (size--) { // 将que1 导入que2,但要留下最后一个元素
- que2.push(que1.front());
- que1.pop();
- }
-
- int result = que1.front(); // 留下的最后一个元素就是要返回的值
- que1.pop();
- que1 = que2; // 再将que2赋值给que1
- while (!que2.empty()) { // 清空que2
- que2.pop();
- }
- return result;
- }
- int top() {
- return que1.back();
- }
- bool empty() {
- return que1.empty();
- }
- };
第二个方法是在第一个方法前提下做优化:
一个队列在模拟栈弹出元素的时候只要将队列头部的元素(除了最后一个元素外) 重新添加到队列尾部,此时再去弹出元素就是栈的顺序了。
上代码加注释:
- class MyStack {
- public:
- queue<int> que;
-
- MyStack() {
-
- }
-
- void push(int x) {
- que.push(x);
- }
-
- int pop() {
- int size = que.size();//记录队列的长度,为了要取出最后一个元素
- size--;//将队列前size-1个元素提取出来就解决了
- while(size--){
- que.push(que.front());//先将首元素提取出来放到队列最后面
- que.pop();//在将首元素弹出
- }
- int result = que.front();//记录首元素(此时是队列的最后一个元素)
- que.pop();//删除
- return result;
- }
-
- int top() {
- return que.back();//队列中的最后一个元素是back
- }
-
- bool empty() {
- return que.empty();
- }
- };
-
- /**
- * 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();
- */
end
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。