使用队列实现栈的下列操作:
push(x) – 元素 x 入栈
pop() – 移除栈顶元素
top() – 获取栈顶元素
empty() – 返回栈是否为空
class MyStack {
public:
/** Initialize your data structure here. */
MyStack() {
listA.clear();
listB.clear();
}
/** Push element x onto stack. */
void push(int x) {
listA.push_back(x);
}
/** Removes the element on top of the stack and returns that element. */
int pop() {
while(!listA.empty() && listA.size() > 1)
{
listB.push_back(listA.front());
listA.pop_front();
}
int numRes = listA.front();
listA.assign(listB.begin(), listB.end());
listB.clear();
return numRes;
}
/** Get the top element. */
int top() {
while(!listA.empty() && listA.size() > 1)
{
listB.push_back(listA.front());
listA.pop_front();
}
return listA.front();
}
/** Returns whether the stack is empty. */
bool empty() {
return (listA.empty() && listB.empty());
}
private:
list<int> listA;
list<int> listB;
};
/**
* 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时先将队列A的最后一个元素以外的元素都拷贝到队列B中,弹出A的最后一个元素,再将B的元素拷贝到A中并清空B;top操作同pop,只是A中保留最后一个元素即可。
使用栈实现队列的下列操作:
push(x) – 将一个元素放入队列的尾部。
pop() – 从队列首部移除元素。
peek() – 返回队列首部的元素。
empty() – 返回队列是否为空。
class MyQueue {
public:
/** Initialize your data structure here. */
MyQueue() {
}
/** Push element x to the back of queue. */
void push(int x) {
staA.push(x);
}
/** Removes the element from in front of queue and returns that element. */
int pop() {
int numRes = -1;
while(staA.size() > 1)
{
staB.push(staA.top());
staA.pop();
}
if(!staA.empty())
{
numRes = staA.top();
staA.pop();
while(!staB.empty())
{
staA.push(staB.top());
staB.pop();
}
}
return numRes;
}
/** Get the front element. */
int peek() {
while(staA.size() > 1)
{
staB.push(staA.top());
staA.pop();
}
return staA.top();
}
/** Returns whether the queue is empty. */
bool empty() {
return staA.empty();
}
private:
stack<int> staA;
stack<int> staB;
};
/**
* 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();
*/
采用两个栈实现队列,pop时将A中最后一个元素以外的元素拷贝到B中,再将A的最后一个元素弹出,并将B中的元素原封不动拷贝到A中;peek同pop操作,保留A中最后一个元素即可。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/implement-queue-using-stacks
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。