题目
一个栈中的元素类型为整型,现在想将该栈从顶到底按从大到小的顺序排序,只许申请一个栈,可以申请新变量,但不能申请额外的数据结构,如何完成排序?
思路
申请辅助栈help,如果从stack出栈的元素大于辅助栈的栈顶元素,那么将辅助栈元素一一弹出并将其重新压入原始栈,直到当前元素小于
辅助栈栈顶或者辅助栈为空时,将当前元素压入。
如果要求从栈顶到栈底从小到大,那么只需判断从stack的元素小于辅助栈的栈顶元素,那么将辅助栈的元素一一弹出并将其重新压入原始栈。具体代码可参考原书
代码
#include<iostream>
#include<algorithm>
#include<stack>
using namespace std;
void sortStackWithStack(stack<int> &s)
{
stack<int> help;
while (!s.empty())
{
int cur = s.top();
s.pop();
while (!help.empty() && cur > help.top())
{
s.push(help.top());
help.pop();
}
help.push(cur);
}
while (!help.empty())
{
s.push(help.top());
help.pop();
}
}
int main()
{
stack<int> input;
for (int i = 3; i < 6; i++)
input.push(i);
for (int i = 1; i < 3; i++)
input.push(i);
sortStackWithStack(input);
for (int i = 0; i < 5; i++)
{
cout << input.top() << endl;
input.pop();
}
getchar();
return 0;
}