双栈排序

题目:

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

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

思路:



代码实现:

class TwoStacks {
public:
    vector<int> twoStacksSort(vector<int> numbers) {
        // write code here
        if(numbers.size() == 0 || numbers.size() == 1)
            return numbers;
        vector<int> result;
        int tmp;
        result.push_back(numbers.back());
        numbers.pop_back();
        while(!numbers.empty())
        {
            tmp = numbers.back();
            numbers.pop_back();
            while(tmp < result.back())
            {
                numbers.push_back(result.back());
                result.pop_back();
            }
            result.push_back(tmp);
        }
        while(!result.empty())
        {
            numbers.push_back(result.back());
            result.pop_back();
        }
        return numbers;
    }
};

猜你喜欢

转载自blog.csdn.net/weixin_40417029/article/details/80973499