华为机试--火车进站

描述

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

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

数据范围：1\le n\le 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]这个序列是不可能实现的。    

这是一道考试栈的题目，其实问题也很明白，就是有一组数，需要你压入栈中，在压入过程中随时可以弹出，但是需要遵守后进先出的原则，也就是1比2先入栈，但是2要比1先出栈才行。

所以这道题的结题步骤是

def dfs(self,wait,stack,leave,ans):
   if len(wait) == 0 and len(stack)== 0:
       ans.append(leave)
   if len(wait) > 0:
       dfs(wait[1:],stack[:]+ [wait[0]],leave)
   if len(stack) > 0:
       dfs(wait[:],stack[:-1],leave+[stack[-1]])
   return ans

这里需要考虑的问题有几个，1，遍历的步骤是什么2，传入的数值有什么要求。

遍历步骤1，设置退出方式如果没有等待的车辆以及栈内也没有车啦，算是一种出栈方式，所以直接算一种，结束。

步骤2.：如果还有车辆在等待，将车辆末尾的数值传入栈中。

步骤3：在这个同时，还有将栈中的数值弹出，所以是if和if并列，而不是if --elif

返回的是收集到的数值。

j解答就是简单啦，只要将需要的数值弄好，就能得到答案，

N = int(input())
train = input().split(" ")
res = dfs(train,[],[],[])
for line in sorted(res):
    print(" ".join(map(str,line)))

当然还能直接用类来设置。

class Solution:
    def __init__(self):
        self.ans = []
    def dfs(self,wait,stack,leave):
        if len(wait) == 0 and len(stack)== 0:
            self.ans.append(leave)
        if len(wait) > 0:
            self.dfs(wait[1:],stack[:]+ [wait[0]],leave)
        if len(stack) > 0:
            self.dfs(wait[:],stack[:-1],leave+[stack[-1]])
        return self.ans

if __name__ == "__main__":
    
    N = int(input())
    train = input().split(" ")
    slu = Solution()
    res = slu.dfs(train,[],[])
    for line in sorted(res):
        print(" ".join(map(str,line)))
    

答案参考的评论中的解答。

DownUp子