面试中常见的堆栈问题

借助一个辅助栈对栈中元素排序

 

题目描述

    一个栈中元素的类型为整形，现在想将该栈从顶到底按从大到小的顺序排序，只许申请一个栈，除此之外，可以申请新的变量，但不能申请额外的数据结构。如何完成排序？

 解题思路

     将栈排序的栈记为arr，申请的辅助栈记为res。在待排序栈arr上执行pop操作，弹出的元素记为stackTop。

    （1）如果stackTop大于或等于res的栈顶元素，则将stackTop直接压入res栈；

    （2）如果stackTop小于res的栈顶元素，则将该res的元素逐个弹出，并压入arr栈，直到辅助栈res的栈顶元素小于待排序栈arr的栈顶元素，再将待排序栈arr的栈顶元素stackTop压入辅助栈res。

由于栈是先进后出，队列是先进先出，而切片可以两头操作，既可以作为栈使用，有可以左右队列使用。下面，使用切片模拟栈的压入弹出来解决该问题：

// sortStackByStack 通过一个辅助栈，对另一个栈进行排序
func sortStackByStack(arr []int) []int {
	res := make([]int, 0)										// 声明辅助栈
	for len(arr) > 0 {											// 当前待排序栈不为空，一直循环
		stackTop := arr[len(arr) - 1]							// 暂存待排序栈顶元素
		arr = arr[:len(arr) - 1]								// 弹出待排序栈顶元素
		for len(res) > 0 && (res[len(res) - 1]) > stackTop {	// 如果辅助栈不为空，并且当前辅助栈栈顶元素大于待排序栈的栈顶元素，那么就一直弹出辅助栈的栈顶元素，并将其压入待排序栈中
			arr = append(arr, res[len(res) - 1])				// 将辅助栈的栈顶元素压入待排序栈中
			res = res[:len(res) - 1]							// 将辅助栈的栈顶元素弹出
		}
		res = append(res, stackTop)								// 如果辅助栈为空，或当前辅助栈栈顶元素小于等于待排序栈的栈顶，就将待排序栈栈顶元素压入辅助栈
	}
	
	return res 
}