- Scala程序员面试算法宝典
- 猿媛之家组编
- 1015字
- 2021-03-23 16:34:52
2.3 如何反转栈的所有元素
【出自ALBB面试题】
难度系数:★★★★☆ 被考察系数:★★★★☆
题目描述:
反转(也叫颠倒)栈的所有元素,如输入栈{1,2,3,4,5}。其中,1处在栈顶,反转之后的栈为{5,4,3,2,1},5处在栈顶。
分析与解答:
最容易想到的办法是,申请一个额外的队列,先把栈中的元素依次出栈放到队列里,然后把队列里的元素按照出队列顺序入栈,这样就可以实现栈的反转,这种方法的缺点是需要申请额外的空间存储队列,因此,空间复杂度较高。下面介绍一种空间复杂度较低的递归的方法。
递归程序有两个关键因素需要注意:递归定义和递归终止条件。经过分析后,很容易得到该问题的递归定义和递归终止条件。递归定义:将当前栈的最底元素移到栈顶,其他元素顺次下移一位,然后对不包含栈顶元素的子栈进行同样的操作。终止条件:递归下去,直到在上图中,对于栈{1,2,3,4,5},进行反转的操作:首先把栈底元素移动到栈顶得到栈{5,1,2,3,4},然后对不包含栈顶元素的子栈进行递归调用(对子栈元素进行反转),子栈{1,2,3,4}反转的结果为{4,3,2,1},因此,最终得到反转后的栈为{5,4,3,2,1}。

此外,由于栈的后进先出的特点,使得只能取栈顶的元素,因此,要把栈底的元素移动到栈顶也需要递归调用才能完成。主要思路:把不包含该栈顶元素的子栈的栈底的元素移动到子栈的栈顶,然后把栈顶的元素与子栈栈顶的元素(其实就是与栈顶相邻的元素)进行交换。

为了更容易理解递归调用,可以认为在进行递归调用时,子栈已经把栈底元素移动到了栈顶。在上图中,为了把栈{1,2,3,4,5}的栈底元素5移动到栈顶,首先对子栈{2,3,4,5},进行递归调用,调用的结果为{5,2,3,4},然后对子栈顶元素5,与栈顶元素1进行交换得到栈{5,1,2,3,4},实现了把栈底元素移动到了栈顶。
实现代码如下:


程序的运行结果为

算法性能分析:
把栈底元素移动到栈顶操作的时间复杂度为O(N),在反转操作中对每个子栈都进行了把栈底元素移动到栈顶的操作,因此,反转算法的时间复杂度为O(N2)。
引申:如何给栈排序。
分析与解答:
很容易通过对上述方法进行修改得到栈的排序算法。主要思路:首先对不包含栈顶元素的子栈进行排序,如果栈顶元素大于子栈的栈顶元素,则交换这两个元素。因此,在上述方法中,只需要在交换栈顶元素与子栈顶元素时增加一个条件判断即可实现栈的排序。
实现代码如下:

程序的运行结果为

算法性能分析:
这种方法的时间复杂度为O(N2)。