232.用栈实现队列
用两个栈来实现队列,一个输入栈,一个输出栈,需要注意的是弹出操作,只有当输出栈为空时,才会一次性把输入栈中的元素都压入输入栈,否则直接取输出栈的元素。
class MyQueue {
public:
stack<int> sIn;
stack<int> sOut;
MyQueue() {
}
//将元素 x 推到队列的末尾
void push(int x) {
sIn.push(x);
}
//从队列的开头移除并返回元素
int pop() {
if(sOut.empty()){
while(!sIn.empty()){
int temp=sIn.top();
sIn.pop();
sOut.push(temp);
}
}
int res=sOut.top();
sOut.pop();
return res;
}
//返回队列开头的元素
int peek() {
int res=this->pop();
sOut.push(res);
return res;
}
//如果队列为空,返回 true ;否则,返回 false
bool empty() {
return sIn.empty()&&sOut.empty();
}
};
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--){//将元素弹入que2,只留下最后一个元素
que2.push(que1.front());
que1.pop();
}
int ans=que1.front;
que1.pop();
que1=que2;//将que2复制给que1
while(!que2.empty()){//将que2清空
que2.pop();
}
return ans;
}
int top() {
que1.back();
}
bool empty() {
return que1.empty();
}
};