AcWing 129. 火车进栈枚举递归 dfs

最新推荐文章于 2023-02-25 23:55:27 发布

Bold!

最新推荐文章于 2023-02-25 23:55:27 发布

阅读量164

点赞数

本文链接：https://blog.csdn.net/srh20/article/details/119189439

版权

这篇博客介绍了如何用深度优先搜索（DFS）解决一个涉及栈操作的数学问题，即有n列火车按特定顺序进站并倒退出站，要求输出所有可能的出栈序列。通过递归实现，博主展示了如何在限制条件下生成前20种字典序排列，并给出了相应的C++代码示例。

摘要由CSDN通过智能技术生成

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

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

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

车站示意如图：

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

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

输入格式

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

输出格式

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

数据范围

1≤n≤20

输入样例：

输出样例：

思路：面对每种状态我们只有两种选择：1.把下一个数进栈 2.把当前栈顶的数出栈。因此我们可以枚举每一步如何选择，递归实现。

代码如下：

#include <bits/stdc++.h>
using namespace  std;
int n,num,top,t,st[26],ans[26];//ans表示出栈的元素 
void dfs(int x){//面对每种状态我们只有两种选择：1.把下一个数进栈 2.把当前栈顶的数出栈 
	if(x==n+1){//递归边界 因为从1开始，所以到n+1结束 
	   if(++num>20) exit(0);//超过20种 
	   for(int i=1;i<=t;i++) printf("%d",ans[i]);
	   for(int j=top;j;j--) printf("%d",st[j]);
	   cout<<endl;
	   return;
	}
	if(top){//把当前栈顶元素出栈 
		ans[++t]=st[top--];
		dfs(x);
		st[++top]=ans[t--];//回溯 
	}
	st[++top]=x;//把当前的数进栈 
	dfs(x+1);
	top--; //回溯 
}
int main(){
	cin>>n;
	dfs(1);
	return 0;
}