/**
请编写一个程序,按升序对栈进行排序(即最大元素位于栈顶),
要求最多只能使用一个额外的栈存放临时数据,
但不得将元素复制到别的数据结构中。
给定一个int[] numbers(C++中为vector<int>),
其中第一个元素为栈顶,请返回排序后的栈。
请注意这是一个栈,意味着排序过程中你只能访问到最后一个元素。
算法:
题目中说只能使用一个额外的栈解决问题,则说明使用的这个额外的栈
就将是最终排序完成的栈。
将输入的待排序的栈记作stack1,最终排序好的栈记作stack2
假设排序过程中的一般情形
stack1包含没有经过排序的N-m个元素,stack2中是已经排序好的m个元素
stack1 stack2
栈顶部
1
5 3
10 8
7 12 栈底部
类似插入排序的思路,stack1栈顶元素就是当前的待排序元素,需要从stack2
中找到合适的位置,将stack1中的元素放入
具体步骤如下:
step1:将stack1栈顶元素弹出,记作temp数值
step2:从stack2中找到合适的位置,(合适的位置是指将temp数值插入到stack2中的位置)
查找算法如下:指针一直指向stack2的栈顶元素,
while(栈顶元素大于temp数值){
将stack2栈顶元素弹出同时append到stack1的栈顶
}
直到stack2的栈顶元素小于或者等于temp数值,while循环结束
step3:
此时stack2的栈顶位置就是将temp数值插入的位置,
再将之前弹出到stack1中的元素都弹回到stack2中,
step4:
当前在关注stack1的栈顶元素,回到步骤1,一直到stack1为空终止
**/
#include<iostream>
using namespace std;
#include<vector>
class TwoStacks {
public:
vector<int> twoStacksSort(vector<int> numbers) {
// write code here
if(numbers.size()<=1){
return numbers;
}
vector<int> stack2;// vector模板类的默认构造函数,创建空向量
while(numbers.size()>0){
int temp=numbers[numbers.size()-1]; //取出stack1栈顶元素
numbers.pop_back();// 删除栈顶元素
int move_time=0;//表示从stack1移动到stack2的元素个数
while(stack2.size()>0){
if(stack2[stack2.size()-1]<temp){
move_time++;
numbers.push_back(stack2[stack2.size()-1]);
// 将stack2栈顶元素添加到stack1栈顶
stack2.pop_back();
}
else{
break;
}
}
// 将temp数值添加到stack2栈顶
stack2.push_back(temp);
// 将move_time个被移到stack1的元素再移动回到stack2
for(int i=0;i<move_time;i++){
stack2.push_back(numbers[numbers.size()-1]);
numbers.pop_back();
}
}
return stack2;
}
};
int main(){
int a[]={1,2,3,4,5};
TwoStacks b;
vector<int> result;
vector<int> in(a,a+5);
result=b.twoStacksSort(in);
for(int i=0;i<result.size();i++){
cout<<result[i]<<endl;
}
return 0;
}