华为OJ—火车进站(栈,字典排序)

时间:2023-03-08 16:01:57
华为OJ—火车进站(栈,字典排序)

http://career-oj.huawei.com/exam/ShowSolution?method=SolutionApp&id=2282

给定一个正整数N代表火车数量,0<N<10,接下来输入火车入站的序列,一共N辆火车,每辆火车以数字1-9编号。要求以字典序排序输出火车出站的序列号。其实也就是输出所有可能的出栈序列。

样例输入:

3

1 2 3

样例输出:

1 2 3
1 3 2
2 1 3
2 3 1
3 2 1




解答:

  其实核心就是一个栈,对于第K个数,在第K个数进栈之前,前面的 K-1 个数要么全部出去了,要么都在栈里面,要么部分在栈里面部分出去了。那么可以假想,在第K个数入栈之前,依次从栈里面出去 0个、1个、2个……栈.size()个,然后把第K个入栈,再对于 K+1个同样实施这样的方法——这就是个递归了。

  出去了的保存在一个队列里面,没出站的保存在栈里面,最后一辆车处理完了递归结束并输出。

代码:

 #include <iostream>
#include <vector>
#include <stack> using namespace std; #define DEBUG int n = ;
int* pArr = NULL; void handle(const int index, stack<int> s, vector<int> v)
{
//对于每一个待处理的数字,先处理栈里面的,再处理这个数字
for (int i = s.size(); i >=; i--)
{
//栈里面的数字可以出来 [0个, 全部],出来的就放到了vector里面待输出了
stack<int> sTemp(s);
vector<int>vTemp(v); //从栈里面出 i 个到队列里面去
for (int j = ; j <= i; j++)
{
int top = sTemp.top();
sTemp.pop();
vTemp.push_back(top);
} //再处理这个,把它放到栈顶
sTemp.push(pArr[index]); if (n - == index)
{
//输出结果
static bool isFirst = true;
vector<int> vRes;
if (!isFirst)
cout << endl; for (int i = ; i < vTemp.size(); i++)
vRes.push_back(vTemp[i]); while (!sTemp.empty())
{
int top = sTemp.top();
sTemp.pop();
vRes.push_back(top);
} int i = ;
for (i = ; i < vRes.size() - ; i++)
cout << vRes[i] << " ";
cout << vRes[i];
isFirst = false;
}
else
{
//递归处理
handle(index + , sTemp, vTemp);
}
}
} int main(void)
{
cin >> n;
pArr = new int[n]; for (int i = ; i < n; i++)
cin >> pArr[i]; stack<int> s;
vector<int> v;
handle(, s, v); delete[] pArr; #ifdef DEBUG
while (true)
cin.get();
#endif
}

第35、36行的变量只是为了输出格式。




总结:

  类似于这种的题目有时候只要根据它的情况模拟一下就知道解决思路了。