按升序对栈进行排序（C++）

题目

编写程序，按升序对栈进行排序（即最大元素位于栈顶）。最多只能使用一个额外的栈存放临时数据，但不得将元素复制到别的数据结构（如数组）。

思路

语言表示欠直观，利用图来说明。（来源《程序员面试金典》）

代码实现

#include <iostream>
#include <exception>
#include <stack>

using namespace std;

// 传入的为引用变量，因为要对stack1中的元素进行修改
void SortStack(stack<int> &stack1)
{
	stack<int> stack2;
	while (!stack1.empty())
	{
		if (stack2.empty())
		{
			int tmp = stack1.top();
			stack2.push(tmp);
			stack1.pop();
		}

		int tmp1 = stack1.top();
		stack1.pop();

		// 若待排序的stack1栈顶元素大于stack2的栈顶元素，则将stack2栈顶不断弹出...
		// 直到stack2栈顶元素小于stack1栈顶元素或stack2为空时
		// 注意：对stack2降序排序，后续将stack2元素转移到stack1中，stack1即为升序排序
		while ((!stack2.empty()) && (stack2.top() < tmp1))
		{
			int tmp = stack2.top();
			stack1.push(tmp);
			stack2.pop();
		}
		stack2.push(tmp1);
	}
	
	// 将stack2中元素转移到stack1中
	while (!stack2.empty())
	{
		int tmp = stack2.top();
		stack1.push(tmp);
		stack2.pop();
	}
}

// 测试代码
int main()
{
	stack<int> stack1;
	stack1.push(3);
	stack1.push(5);
	stack1.push(1);
	stack1.push(9);
	stack1.push(7);

	SortStack(stack1);
	
	// 验证stack1被正确排序
	while (!stack1.empty())
	{
		int tmp = stack1.top();
		cout << tmp << endl;
		stack1.pop();
	}

	return 0;
}
复制代码

参考文献

[1] 队列和栈面试题