通过两个队列来模拟一个堆栈的简单行为:
实现堆栈的压入和弹出功能:
//*********用两个队列实现一个栈*************
//思路
//1.初始化,我们要对堆栈进行压入操作,当两个队列都为空,将元素插入任意一个队列
//2.当要从堆栈中弹出时,要删除有元素的队列的队尾元素,通过将不等于将要删除元素的
//队列中的元素依次插入另一个队列。并删除队尾元素(弹出元素)
//3.如果再次对堆栈进行压入操作,往队列不为空的那个队列进行插入
//4.堆栈为空,弹出元素,显示错误
class Solution2
{
public:
void push(int node) {
if (!queue1.empty())
queue1.push(node);
else
queue2.push(node);
}
int pop() {
int pop_data;
if (!queue1.empty())
{
pop_data=queue1.back();
while(!queue1.empty())
{
int insert_data=queue1.front();
if (insert_data!=pop_data)
queue2.push(insert_data);
queue1.pop();
}
}
else if(!queue2.empty())
{
pop_data=queue2.back();
while(!queue2.empty())
{
int insert_data=queue2.front();
if (insert_data!=pop_data)
queue1.push(insert_data);
queue2.pop();
}
}
else
throw runtime_error("no data in stack");
return pop_data;
}
private:
queue<int> queue1;
queue<int> queue2;
};
//往空的栈里添加和删除元素
void Test1()
{
cout<<"Test1 begin: ";
Solution2 sol;
sol.push(1);
sol.push(2);
cout<<sol.pop()<<endl;
sol.push(3);
cout<<sol.pop()<<endl;
sol.push(4);
cout<<sol.pop()<<endl;
sol.push(5);
cout<<sol.pop()<<endl;
cout<<sol.pop()<<endl;
cout<<endl;
}
//连续删除元素直至栈为空
void Test2()
{
cout<<"Test2 begin: ";
Solution2 sol;
sol.push(1);
sol.push(2);
sol.push(3);
sol.push(4);
sol.push(5);
cout<<sol.pop()<<endl;
cout<<sol.pop()<<endl;
cout<<sol.pop()<<endl;
cout<<sol.pop()<<endl;
cout<<sol.pop()<<endl;
cout<<sol.pop()<<endl;//非法操作空栈
}
int _tmain(int argc, _TCHAR* argv[])
{
Test1();
Test2();
return 0;
}