题目要求:一个栈中元素类型为整型,现在想将该栈自顶到底从大到小排序(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;
}
};