Cracking the coding interview--Q3.6

题目

原文：

Write a program to sort a stack in ascending order. You should not make any assumptions about how the stack is implemented. The following are the only functions that should be used to write this program: push | pop | peek | isEmpty.

译文：

写一个程序对一个栈进行升序排序，对栈是如何实现的，你不用任何假想，程序能用到的函数有：push，pop，peek，isEmpty

解答

使用一个附加的栈来模拟插入排序。将原栈中的数据依次出栈与附加栈中的栈顶元素比较， 如果附加栈为空，则直接将数据压栈。否则， 如果附加栈的栈顶元素大于从原栈中弹出的元素，则将附加栈的栈顶元素压入原栈。 一直这样查找直到附加栈为空或栈顶元素已经不大于该元素， 则将该元素压入附加栈。这样类似插入排序的方法，之间复杂度为O(n^2)。

代码如下：

import java.util.*;class Q3_6{public static Stack<Integer> sort(Stack<Integer> s) {Stack<Integer> r = new Stack<Integer>();while(!s.isEmpty()) {int tmp = s.pop();while(!r.isEmpty() && r.peek() > tmp) {s.push(r.pop());}r.push(tmp);}return r;}public static void main(String[] args){int[] a={3,6,2,9,7};Stack<Integer> s=new Stack<Integer>();for(int i=0;i<a.length;i++)s.push(a[i]);s=sort(s);while(!s.isEmpty()){System.out.print(s.pop()+" ");}}}

---EOF---