题目1:232 用栈实现队列
题目链接:232 用栈实现队列
题意
用两个栈实现先入先出队列(一个入栈,一个出栈),实现如下功能:
1)push:将元素x推到队列末尾
2)pop:从队列的开头移除并返回元素
3)peek:返回队列开头的元素
4)empty:若队列为空,返回true,否则,返回false
代码
class MyQueue {
public:
stack<int> stackIn;//入栈
stack<int> stackOut;//出栈
MyQueue(){}
void push(int x){
stackIn.push(x);
}
int pop(){
//stackOut出栈为空时,放入元素
if(stackOut.empty()){
while(!stackIn.empty()){
stackOut.push(stackIn.top());
stackIn.pop();
}
}
//出栈不为空时,直接弹出元素
int result = stackOut.top();
stackOut.pop();
return result;
}
int peek(){
int result = this->pop();//复用上面的pop()函数,
stackOut.push(result);//但是还需要将元素放回出栈中
return result;
}
bool empty(){
return (stackIn.empty() && stackOut.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();
*/
- 时间复杂度: push和empty为O(1), pop和peek为O(n)
- 空间复杂度: O(n)
题目2: 225 用队列实现栈
题目链接:225 用队列实现栈
题意
使用两个队列实现栈,实现如下功能
push:将元素x压入栈顶
pop:移除并返回栈顶的元素
top:返回栈顶的元素
empty:栈为空,返回true,否则,返回false
两个队列
其中一个队列(que2)用来备份,把que1要弹出的元素以外的元素都备份到que2,然后弹出que1中的那个元素,再将que2中的元素放到que1中,同时清空que2
逻辑
例1:que2每次都要清空
每pop一次,que2都要备份一次,一定要是空的,才能接续不断地进行操作,如果不清空的话,有可能已经弹出的元素会再次回到栈中
例2:que2的全部元素都要移动到que1中
因为que2中保存的是当前pop操作que1中没有用到的元素,为了保证后续操作,要将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--){
que2.push(que1.front());//que2备份que1弹出的元素
que1.pop();
}
int result = que1.front();
que1.pop();
//que1 = que2
while(!que2.empty()){
que1.push(que2.front());
que2.pop();
}
return result;
}
int top(){
return que1.back();
}
bool empty(){
return que1.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();
*/
- 时间复杂度: pop为O(n),其他为O(1)
- 空间复杂度: O(n)
一个队列(★)
模拟出栈时,将队列头部(出)的size-1个元素依次重新添加到队尾(入),剩下的那个没有移动的元素就是所求
代码
class MyStack {
public:
queue<int> que;
MyStack(){}
void push(int x){
que.push(x);
}
int pop(){
int size = que.size();
size--;
while(size--){
que.push(que.front());
que.pop();
}
int result = que.front();
que.pop();
return result;
}
int top(){
return que.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();
*/
- 时间复杂度: pop为O(n),其他为O(1)
- 空间复杂度: O(n)