[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 = 0, tmp;
        while (top != numbers.size()) {
            tmp = numbers[top++];
            while (!stk.empty() && stk.top() > tmp) {
                numbers[--top] = stk.top();
                stk.pop();
            }
            stk.push(tmp);
        }
        top = 0;
        while (!stk.empty()) {
            numbers[top++] = stk.top();
            stk.pop();
        }
        return numbers;
    }
};