Q3.5 Implement a MyQueue class which implements a queue using two stacks.

最新推荐文章于 2022-12-23 17:56:08 发布

原创最新推荐文章于 2022-12-23 17:56:08 发布 · 562 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Stack

Cracking the coding interview 专栏收录该内容

50 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现队列的方法。通过这种方式，可以有效地模拟队列的先进先出特性。主要实现了push、pop、front和back等基本队列操作。

Q:Implement a MyQueue class which implements a queue using two stacks.

A:一个栈sin用于进，一个栈sout用于出。

push：直接push进sin即可

pop：如果sout不空，直接pop，否则将sin中的元素都压入sout之后再sout.pop

front: 同pop，只不过不pop，而是top

back：同front，如果sin不为空，直接sin.top()，否则将sout元素都压入sin, 再sin.pop

#include <iostream>
#include <stack> 
using namespace std;

template <typename T>
class MyQueue{
	private:
		stack<T> sin, sout;
	
	public:
		MyQueue() {
		}
		
		~MyQueue() {
		}
		
	void push(T val){
		sin.push(val);
    }
    
    void pop(){
    	move(sin, sout);
    	sout.pop();
    }
    
    T front(){
		move(sin, sout);
		return sout.top();
    }
    
    T back(){
		move(sout, sin);
		return sin.top();
    }
    
    int size(){
		return sin.size()+sout.size();
    }
    
    bool empty(){
		return sin.empty()&&sout.empty();
    }
    
    void move(stack<T> &src, stack<T> &dst){
    	while (dst.empty()) {
    		while (!src.empty()) {
    			dst.push(src.top());
    			src.pop();
    		}
    	}
    }
};

int main() {
	MyQueue<int> q;
    MyQueue<int> q1;
    for(int i=0; i<10; ++i){
        q.push(i);
        q1.push(i);
    }

    cout<<q.front()<<" "<<q.back()<<endl;
    cout<<q1.front()<<" "<<q1.back()<<endl;
    cout<<endl;
    q.pop();
    q1.pop();
    q.push(10);
    q1.push(10);
    cout<<q.front()<<" "<<q.back()<<endl;
    cout<<q1.front()<<" "<<q1.back()<<endl;
    cout<<endl;
    cout<<q.size()<<" "<<q.empty()<<endl;
    cout<<q1.size()<<" "<<q1.empty()<<endl; 
	return 0;
}