AcWing 129. 火车进栈

AcWing 129. 火车进栈

这里有 n 列火车将要进站再出站，但是，每列火车只有 1 节，那就是车头。

这 n 列火车按 1 到 n 的顺序从东方左转进站，这个车站是南北方向的，它虽然无限长，只可惜是一个死胡同，而且站台只有一条股道，火车只能倒着从西方出去，而且每列火车必须进站，先进后出。

也就是说这个火车站其实就相当于一个栈，每次可以让右侧头火车进栈，或者让栈顶火车出站。

车站示意如图：

      	  出站<——    <——进站
                 |车|
                 |站|
                 |__|

现在请你按《字典序》输出前 20 种可能的出栈方案。

输入格式
输入一个整数 n，代表火车数量。

输出格式
按照《字典序》输出前 20 种答案，每行一种，不要空格。

数据范围
1≤n≤20

输入样例：
3
输出样例：
123
132
213
231
321

思路：

        state1 					state3
        出站<——   				<——进站
                    |车|
                    |站|
                    |__|
                   state2

state1记录的是：作为已经出栈的火车序列，是最后输出的结果，是一个vector
state2：作为车站的栈，保存现在在栈内的火车序列，是一个栈
state3: 作为尚未入栈的火车序列，是一个数字

用dfs,先枚举从车站到出站的情况，也就是从state2到state1的情况。
再枚举进站到车站的情况，也就是从state3到state2的情况。

这样枚举可以保证前面的元素比后面的小，因为后面的车号比前面的大，如果前面的先出站，后面的车再入栈然后出站，可以保证字典序。

#include <bits/stdc++.h>
using namespace std;

int n,cnt=20;//cnt表示最多可列举20种情况，n表示共有多少车
vector<int>state1;
stack<int> state2;
int state3=1;//车入栈的第一辆车是序号为1

void dfs(){
	if(!cnt) return ;//如果20次已经输出完了
	if(state1.size()==n){//如果所有车都已出站
		cnt--;//可列举的次数减一
		for(auto x:state1) cout<<x;
		cout<<endl;
		return;
	}
	
	//先枚举从车站到出站的情况，也就是从state2到state1的情况。
	if(state2.size()){//如果栈内还有车
		state1.push_back(state2.top());//车出站，出站的车为栈顶元素
		state2.pop();
		dfs();
		state2.push(state1.back());//回溯
		state1.pop_back();
	}
	
	//再枚举进站到车站的情况，也就是从state3到state2的情况。
	if(state3<=n){
		state2.push(state3);
		state3++;
		dfs();
		state3--;
		state2.pop();
	}
} 
int main() {
	cin>>n;
	dfs();
  return 0;
}

此题的题解视频