题目
一个栈依次压入1、2、3、4、5,那么从栈顶到栈底分别为5、4、3、2、1,将这个栈转置后,从栈顶到栈底为1、2、3、4、5,也就是实现栈中元素的逆序,但是只能用递归函数来实现,不能用其他数据结构
分析
可以用两个递归函数
int getAndRemoveLastElement(Stack stack)
这个函数的作用是:获取并移出这个栈的栈底元素。这个函数实现分情况,当栈只有一个元素时,弹出的栈顶元素即为栈底元素,当多于一个元素时,先弹出栈顶元素,使得栈的size减少,然后对该栈取栈底元素(递归),再把原栈顶元素放回,返回栈底元素。不会有元素少于一个的情况,因为数量等于1时直接返回栈顶,没有继续递归,stack的size为0时也不会调用到这个函数。
void reverse(Stack stack)
函数作用:反转栈。先取出栈底元素,然后反转剩下的栈,把栈底元素压入栈顶即可。
代码实现
import java.util.Stack;
public class ReverseStack {
private int getAndRemoveLastElement(Stack<Integer> stack){
int val = stack.pop();
if (stack.empty()){
return val;
} else {
int last = getAndRemoveLastElement(stack);
stack.push(val);
return last;
}
}
public void reverse(Stack<Integer> stack){
if(!stack.empty()){
int val = getAndRemoveLastElement(stack);
reverse(stack);
stack.push(val);
}
}
public static void main(String[] args) {
Stack<Integer> stack = new Stack<>();
stack.push(1);
stack.push(2);
stack.push(3);
stack.push(4);
stack.push(5);
System.out.println(stack);
new ReverseStack().reverse(stack);
System.out.println(stack);
}
}
参考于《程序员代码面试指南》第二版