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---