题目
July 8, 2021 · View on GitHub
这里有 n 列火车将要进站再出站,但是,每列火车只有 1 节,那就是车头。
这 n 列火车按 1 到 n 的顺序从东方左转进站,这个车站是南北方向的,它虽然无限长,只可惜是一个死胡同,而且站台只有一条股道,火车只能倒着从西方出去,而且每列火车必须进站,先进后出。
也就是说这个火车站其实就相当于一个栈,每次可以让右侧头火车进栈,或者让栈顶火车出站。
车站示意如图:
出站<—— <——进站
|车|
|站|
|__|
现在请你按《字典序》输出前 20 种可能的出栈方案。
输入格式
输入一个整数 n,代表火车数量。
输出格式
按照《字典序》输出前 20 种答案,每行一种,不要空格。
数据范围
1≤n≤20
输入样例:
3
输出样例:
123
132
213
231
321
参考答案
//
// Created by Genes on 2020/12/19.
//
// 火车进栈 dfs
// 面对任何一个状态, 我们只有两种选择:
// 1. 把下一个数进栈
// 2. 把当前栈顶的数出栈(如果栈非空)
#include<iostream>
#include<vector>
using namespace std;
const int N = 2e1 + 5;
int n, remain = 0;
int tt, stk[N];
vector<int> path;
void dfs(int u) {
if (u == n + 1) { // edge
if (++remain > 20) {
exit(0);
}
for (auto x:path) { //输出路径中的
cout << x;
}
// 如果没有数字可用了, 那么输出一下path+栈中的数字
for (int i = tt - 1; i >= 0; i--) { //输出栈中剩下的(后进先出
cout << stk[i];
}
cout << endl;
return;
}
if (tt) {
path.push_back(stk[--tt]);
dfs(u);
stk[tt++] = path.back();
path.pop_back();
}
stk[tt++] = u;
dfs(u + 1);
tt--;
}
int main() {
cin >> n;
dfs(1);
return 0;
}