题目描述:
思路一:
使用两个队列,压入 -O(1), 弹出 -O(n)
每次压栈时直接push();出栈时弹出队列尾部
代码实现:
class MyStack {
private:
queue<int> q1;
queue<int> q2;
int to;
public:
/** Initialize your data structure here. */
MyStack() {
}
/** Push element x onto stack. */
void push(int x) {
q1.push(x);
to=x;
}
/** Removes the element on top of the stack and returns that element. */
int pop() {
int res;
while(q1.size()!=1){
q2.push(q1.front());
q1.pop();
}
res=to;
q1.pop();
while(q2.size()!=0){
to=q2.front();
q1.push(q2.front());
q2.pop();
}
return res;
}
/** Get the top element. */
int top() {
return to;
}
/** Returns whether the stack is empty. */
bool empty() {
if(q1.size()==0){
return true;
}else{
return false;
}
}
};
执行结果:
思路二:
使用两个队列,压入 -O(n), 弹出 -O(1)
每次压栈时,在队列中逆序排列;出栈时直接pop()
代码实现:
class MyStack {
private:
queue<int> q1;
queue<int> q2;
int to;
public:
/** Initialize your data structure here. */
MyStack() {
}
/** Push element x onto stack. */
void push(int x) {
q2.push(x);
to=x;
while(q1.size()!=0){
q2.push(q1.front());
q1.pop();
}
while(q2.size()!=0){
q1.push(q2.front());
q2.pop();
}
}
/** Removes the element on top of the stack and returns that element. */
int pop() {
int res=to;
q1.pop();
to=q1.front();
return res;
}
/** Get the top element. */
int top() {
return to;
}
/** Returns whether the stack is empty. */
bool empty() {
if(q1.size()==0){
return true;
}else{
return false;
}
}
};
执行结果:
思路三:
使用一个队列,压入 - O(n), 弹出 - O(1)
压入时进行q.size()-1次push()和pop(),倒序排列;弹出时直接pop()
代码实现:
class MyStack {
private:
queue<int> q;
int to;
public:
/** Initialize your data structure here. */
MyStack() {
}
/** Push element x onto stack. */
void push(int x) {
q.push(x);
to=x;
int sz=q.size()-1;
while(sz){
q.push(q.front());
q.pop();
sz--;
}
}
/** Removes the element on top of the stack and returns that element. */
int pop() {
int res=to;
q.pop();
to=q.front();
return res;
}
/** Get the top element. */
int top() {
return to;
}
/** Returns whether the stack is empty. */
bool empty() {
if(q.size()==0){
return true;
}else{
return false;
}
}
};