【死磕算法·栈和队列】双栈排序问题

题目要求:一个栈中元素类型为整型,现在想将该栈自顶到底从大到小排序(vector中第一个元素为栈顶),只能申请一个栈,除此之外可以申请新的变量,但不能申请额外的数据结构,如何完成排序?

思路:记原来的栈为stack,申请的辅助栈为help,设置help自顶到底从小到大排序,即可将help倒入stack中,实现stack自顶到底从大到小排序。

从stack中弹出栈顶元素,记为current。

如果current小于等于help的栈顶元素,将current压入help中;

如果current大于help的栈顶元素,将help中的元素逐一弹出压到stack中,直到current小于等于help的栈顶元素。

直到stack中的元素全部转移到了help中,将help的栈元素逐一压回stack中即可。

此题用vector实现stack,vector的第一个元素为栈顶,则在待返回的vector中vector[0]为最大值,则申请的辅助vector中应该是vector[0]是最小值.

因此在help 向量中 vector[0]...vector[length-1] 从小到大排序;

numbers向量中numbers[0]...numbers[lenght-1]从大到小排序。

class TwoStacks {
public://用vector模拟栈操作
    vector<int> twoStacksSort(vector<int> numbers) {
        // write code here
        vector<int> help;
        while(!numbers.empty()){
             int current = numbers.back();
             numbers.pop_back();
             if(help.empty())
                 help.push_back(current);
             else{
                 if(current>=help.back())
                     help.push_back(current);
                 else{
                     while(!help.empty()&& current<help.back()){
                         numbers.push_back(help.back());
                         help.pop_back();
                     }
                     help.push_back(current);
                 }
                 
                 
              }
        }
        while(!help.empty()){
             numbers.push_back(help.back());
             help.pop_back();
        }
        return numbers;
            }
};

猜你喜欢

转载自blog.csdn.net/gulaixiangjuejue/article/details/85040229