牛客-华为机试-中等-栈-HJ77火车进站-C++?

描述

给定一个正整数N代表火车数量，0<N<10，接下来输入火车入站的序列，一共N辆火车，每辆火车以数字1-9编号，火车站只有一个方向进出，同时停靠在火车站的列车中，只有后进站的出站了，先进站的才能出站。

要求输出所有火车出站的方案，以字典序排序输出。

数据范围：1≤n≤10 1≤n≤10 

进阶：时间复杂度：O(n!) O(n!) ，空间复杂度：O(n) O(n) 

输入描述：

第一行输入一个正整数N（0 < N <= 10），第二行包括N个正整数，范围为1到10。

输出描述：

输出以字典序从小到大排序的火车出站序列号，每个编号以空格隔开，每个输出序列换行，具体见sample。

示例1

输入：

3
1 2 3

输出：

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

说明：

第一种方案：1进、1出、2进、2出、3进、3出
第二种方案：1进、1出、2进、3进、3出、2出
第三种方案：1进、2进、2出、1出、3进、3出
第四种方案：1进、2进、2出、3进、3出、1出
第五种方案：1进、2进、3进、3出、2出、1出
请注意，[3,1,2]这个序列是不可能实现的。     

代码

#include<iostream>
#include<vector>
#include<stack>
#include<algorithm>
using namespace std;
int train[10];
int n;
vector<string>paths;
vector<int>path;
stack<int>stk;
void dfs(int i){
    if(path.size()==n){
        string s(n*2-1,' ');
            for(int i=0;i<n;i++)
                s[i*2]=('0'+path[i]);
            paths.push_back(s);
            return;
    }
    if(!stk.empty()){
        path.push_back(stk.top());
        stk.pop();
        dfs(i);
        stk.push(path.back());
        path.pop_back();
    }
    if(i<n){
        stk.push(train[i]);
        dfs(i+1);
        stk.pop();
    }
}
int main(){
    cin>>n;//输入n个元素
    for(int i=0;i<n;i++)
        cin>>train[i];//输入火车数组
    dfs(0);
    sort(paths.begin(),paths.end());
    for(auto&s:paths)
        cout<<s<<'\n';
    return 0;
}