stackでqueueを実現-これはよくある面接問題です.

1590 ワード

自分がstackで実現したqueueは、いくつかの基本的なテストに合格することができます.もし皆さんがその中で間違いを見つけることができたら、指摘したいことがあります.
#include<iostream>
#include<cassert>
#include<stack>
using namespace std;
template <class T>
class Queue{
private:
	T back_data;
	stack<T> m_stack1;
	stack<T> m_stack2;
public:
	Queue() {
	}
	~Queue() {
	}
	void push_back(int data);
	void pop_front();
	T front();
	T back();
};

template <class T>
void Queue<T>::push_back(int data) {
	m_stack1.push(data);
	back_data = data;
}

template <class T>
void Queue<T>::pop_front() {
	if(m_stack2.size() == 0) {
		int temp;
		while(!m_stack1.empty()) {
			temp = m_stack1.top();
			m_stack1.pop();
			m_stack2.push(temp);
		}
	}
	if(m_stack2.size() > 0) {
		m_stack2.pop();
	}
}

template <class T>
T Queue<T>::back() {
	return back_data;
}

template <class T> 
T Queue<T>::front() {
	if(m_stack2.size() == 0) {
		int temp;
		while(!m_stack1.empty()) {
			temp = m_stack1.top();
			m_stack1.pop();
			m_stack2.push(temp);
		}
	}
	assert(m_stack2.size() > 0);
	return m_stack2.top();
}
void main() {
	Queue<int> m_queue;
	m_queue.push_back(2);
	m_queue.push_back(3);
	cout << m_queue.back() << endl;
	cout << m_queue.front() << endl;
	m_queue.pop_front();
	cout << m_queue.front() << endl;
	m_queue.pop_front();
	m_queue.push_back(4);
	m_queue.push_back(5);
	m_queue.push_back(6);
	cout << m_queue.front() << endl;
	m_queue.pop_front();
	cout << m_queue.front() << endl;
}