火车出站 模拟栈

这是我记不清原题的一道题，大致题意如下:
有n辆火车，编号1~n，进站顺序为1~n，进站和出站符合栈的规律：必须让最后进站的车先出站，并且在进站的同时也可以出站。请输出所有出站的可能方式和方案总数。按字典序递增输出。
样例
3
1 2 3
1 3 2
2 1 3
2 3 1
3 2 1
5

找不到地方提交，贴下我的代码：

#include<cstdio>
#include<cstring>
const int maxn=1e3;
int vis[maxn],ans[maxn];
int n,cnt=0;
void dfs(int now,int b,int id){
    if(now==n){
        cnt++;
        for(int i=n;i>0;--i)
            if(!vis[i]) ans[id++]=i;
        for(int i=0;i<n;++i){
            if(i==0) printf("%d",ans[i]);
            else printf(" %d",ans[i]);
        }
        printf("\n");
        return;
    }
    int old[maxn],p;
    for(int i=b;i>=0;--i){ //返回打印从大到小才能保证以字典序输出
        int t=i,id1=id; p=0;
        for(int j=now-1;t>0;--j){ //从now位置出站
            if(!vis[j]) {
                old[p++]=j; vis[j]=1; ans[id1++]=j;
                --t;
            }
        }
        for(int k=1;k<=n;++k){
            if(now+k>n) break;
            vis[now+k]=1;
            ans[id1]=now+k;
            dfs(now+k,b+k-i-1,id1+1);
            vis[now+k]=0;
        }
        for(int k=0;k<p;++k) vis[old[k]]=0;
    }
    return;
}
int main(){
    while(scanf("%d",&n)==1&&n){
        memset(vis,0,sizeof(vis));
        cnt=0;
        dfs(0,0,0);
        printf("%d\n",cnt);
    }
}

如有不当之处欢迎指出！！

转载于:https://www.cnblogs.com/flyawayl/p/8305572.html