程序员面试金典-3.6双栈排序

一、题目描述

请编写一个程序，按升序对栈进行排序（即最大元素位于栈顶），要求最多只能使用一个额外的栈存放临时数据，但不得将元素复制到别的数据结构中。

给定一个int[] numbers(C++中为vector<int>)，其中第一个元素为栈顶，请返回排序后的栈。请注意这是一个栈，意味着排序过程中你只能访问到第一个元素。

测试样例：

[1,2,3,4,5]

返回：[5,4,3,2,1]

二、题目思路

该思路是参考论坛里面某位大神的做法，创建两个栈，一个初始栈，一个排序栈。首先将numbers里面的元素全部添加到初始栈，然后弹出初始栈的栈顶元素按照以下规则添加到排序栈中。目标是生成排序栈中元素从栈底到栈顶的正序排列

1.如果排序栈为空或者排序栈顶元素小于初始栈顶元素，初始栈中的元素可以直接添加到排序栈。

2.如果排序栈的栈顶元素大于初始栈顶元素，则弹出初始栈顶元素为变量popint，然后把排序栈中大于popint的元素添加到初始栈中，然后将popin压入排序栈，最后将刚才压入初始栈中的元素再压入排序栈。

三、代码

import java.util.*;

public class TwoStacks {
    public ArrayList twoStacksSort(int[] numbers) {
        // write code here
        Stack bufer=new Stack();
        Stack sorted=new Stack();
        ArrayList result=new ArrayList();
        for(int i=0;i<=numbers.length-1;i++)
            bufer.push(numbers[i]);
        //sorted.push(numbers[numbers.length-1]);
        int popint=0;
        while(!bufer.empty()){
            if(sorted.empty()||sorted.peek()<=bufer.peek())
                sorted.push(bufer.pop());
            else{
                int count=0;
                popint=bufer.pop();
                while(!sorted.empty() && sorted.peek()>popint){
                    bufer.push(sorted.pop());
                    count++;
                }
                sorted.push(popint);
                for(int i=0;i