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;
}