题目描述
给你一个1到n的排列和一个栈,入栈顺序给定,你要在不打乱入栈顺序的情况下,对数组进行从大到小排序。当无法完全排序时,请输出字典序最大的出栈序列
示例1
输入
[2,1,5,3,4]
返回值
[5,4,3,1,2]
题解:乍看一眼,涉及到栈的出栈和入栈问题,要想不打破入栈顺序,就要手工模拟一个栈出来,不能直接使用数组来得到最终的结果。获取字典序最大的出栈序列是本题目的一个限制,根据题目中给定的 n 就可以得到理想的字典序最大的序列,但是需要结合真正的出栈序列来考虑。要得到最大的字典序列那么第一位数字就是 n ,理想情况下第二位是 n-1。首先确定弹栈的方式,使用一个标记数组用于标记当前该位是否遍历过,当然是从 n 开始逆序遍历直到找到第一个未访问过的数字停止,栈顶元素大于等于 n 即可弹栈,放到最终的结果序列里面,如果数组元素遍历完,栈内无剩余元素即表示得到的是一个逆序的序列,否则表示的是一个普通的序列,只不过在出栈顺序下字典序最大。
class Solution {
public:
/**
* 栈排序
* @param a int整型一维数组 描述入栈顺序
* @param aLen int a数组长度
* @return int整型vector
*/
vector<int> solve(int* a, int aLen) {
// write code here
stack<int> stk;
int n = aLen;
vector<int> res;
vector<bool> flag(n+1,false);
for(int i = 0; i < aLen; i++)//使用固定的变量aLen,否则没有遍历完就结束了
{
stk.push(a[i]);
flag[a[i]]=true;
while(n && flag[n]) n--;
while(!stk.empty() && n<=stk.top())
{
int t = stk.top();
res.push_back(t);
stk.pop();
}
}
while(!stk.empty())
{
int t = stk.top();
res.push_back(t);
stk.pop();
}
return res;
}
};