[CTCI] 双栈排序

双栈排序

题目描述

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

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

测试样例：

[1,2,3,4,5]

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

 class TwoStacks {

 public:

     vector<int> twoStacksSort(vector<int> numbers) {

         // write code here

         stack<int> stk;

         int top = , tmp;

         while (top != numbers.size()) {

             tmp = numbers[top++];

             while (!stk.empty() && stk.top() > tmp) {

                 numbers[--top] = stk.top();

                 stk.pop();

             }

             stk.push(tmp);

         }

         top = ;

         while (!stk.empty()) {

             numbers[top++] = stk.top();

             stk.pop();

         }

         return numbers;

     }

 };