有的同学可能疑惑这种题目有什么实际工程意义,其实很多算法题目主要是对知识点的考察和教学意义远大于其工程实践的意义,以下两题便是如此
首先,先牢记栈的四个基本操作:
- push(x) – 将一个元素放入栈的顶部
- pop() – 从栈的顶部移除元素
- peek() – 返回栈顶部的元素
- empty() – 返回栈是否为空
队列的四个基本操作(一样的):
- push(x) – 将一个元素放入队列的尾部
- pop() – 从队列首部移除元素
- peek() – 返回队列首部的元素
- empty() – 返回队列是否为空
232. 用栈实现队列
这是一道模拟题,不涉及到具体算法,考察的就是对栈和队列的掌握程度。
使用栈来模式队列的行为,如果仅仅用一个栈,是一定不行的,所以需要两个栈一个输入栈,一个输出栈,这里要注意输入栈和输出栈的关系
在push数据的时候,只要数据放进输入栈就好,但在pop的时候,操作就复杂一些,输出栈如果为空,就把进栈数据全部导入进来(注意是全部导入),再从出栈弹出数据,如果输出栈不为空,则直接从出栈弹出数据就可以了
最后如何判断队列为空呢?如果进栈和出栈都为空的话,说明模拟的队列为空了
class MyQueue {
public:
stack<int> stIn;
stack<int> stOut;
MyQueue() {
}
void push(int x) {
stIn.push(x);
}
int pop() {
//只有当stOut为空的时候,再从stIn里导入全部数据
//否则直接读取stOut的第一个数据即可
if(stOut.empty()){
//从stIn里导入数据直到stIn为空
while(!stIn.empty()){
stOut.push(stIn.top());
stIn.pop();
}
}
int result = stOut.top(); //让result等于栈顶
stOut.pop(); //移除栈顶
return result;
}
int peek() {
int res = this->pop(); //直接使用已有的pop函数;
stOut.push(res); //因为使用pop弹出了栈顶元素,所以需要再添加回去
return res;
}
bool empty() {
return stIn.empty() && stOut.empty();
}
};
- 时间复杂度: push和empty为O(1), pop和peek为O(n)
- 空间复杂度: O(n)
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; //再将赋值给que1
while(!que2.empty()){ //清空que2
que2.pop();
}
return result;
}
int top() {
return que1.back();
}
bool empty() {
return que1.empty();
}
};
- 时间复杂度: pop为O(n),其他为O(1)
- 空间复杂度: O(n)
理解了以上思路,可以进行优化,只用一个队列实现栈功能
在模拟栈弹出元素的时候,只要将队列头部的元素(除了最后一个)重新添加到队列尾部,此时再弹出元素的时候就是栈的顺序了
class MyStack {
public:
queue<int> que1;
MyStack() {
}
void push(int x) {
que1.push(x);
}
int pop() {
int size = que1.size();
size--;
while(size--){ //将队列头部的元素(除了最后一个)重新添加到队列尾部
que1.push(que1.front());
que1.pop();
}
int result = que1.front(); //此时再弹出元素的时候就是栈的顺序了
que1.pop();
return result;
}
int top() {
return que1.back();
}
bool empty() {
return que1.empty();
}
};
- 时间复杂度: push和empty为O(1), pop和peek为O(n)
- 空间复杂度: O(n)
题外话
可以看出peek()的实现,直接复用了pop(), 要不然,对stOut或者que判空的逻辑又要重写一遍
在工业级别代码开发中,最忌讳的就是实现一个类似的函数,直接把代码粘过来改一改就完事了
这样的项目代码会越来越乱,当这段代码需要修改的时候,复制了n次就要修改n次。一定要懂得复用,功能相近的函数要抽象出来,不要大量的复制粘贴,很容易出问题!