题目描述
宁宁考虑的是这样一个问题:一个操作数序列:1,2,…,n(图示为 1 到 3 的情况),栈 A 的深度大于 n。
现在可以进行两种操作,
- 将一个数,从操作数序列的头端移到栈的头端(对应数据结构栈的 push 操作)
- 将一个数,从栈的头端移到输出序列的尾端(对应数据结构栈的 pop 操作)
使用这两种操作,由一个操作数序列就可以得到一系列的输出序列,下图所示为由 1 2 3
生成序列 2 3 1
的过程。
(原始状态如上图所示)
你的程序将对给定的 nn,计算并输出由操作数序列 1,2,…,n 经过操作可能得到的输出序列的总数。
这个题目最难的其实是搞清本质,不要被一堆数迷惑了。先想一个问题,数是多少或者哪个数先进栈出栈有影响吗?
没有影响。因为要输出的只是方案数,哪个数先出栈都无所谓。输出序列里怎么排都无所谓。
当栈里有x个数,队列里有y个数时:此时所有方案数包含两种情况:一个数出栈或者从队列里拿一个数进栈,也就是(x,y) = 出栈(x-1,y)+ 进栈(x+1)+(y-1)
n = int(input())
dp = [[0]*19 for i in range(19)] # 建立一个二位列表,存放当前方案数
def DFS(x,y):
if y == 0: # 队列里没数了,都在输出队列里或者栈里,此时就一种方案
return 1
if dp[x][y]: # 如果前面已经算过了(x,y)时的方案数,直接返回值
return dp[x][y]
if x: # 出栈 只有栈不是空的才能从栈里拿数
dp[x][y] += DFS(x-1,y)
dp[x][y] += DFS(x+1,y-1) # 进栈
return dp[x][y]
DFS(0,n)
print(dp[0][n])
代码很简单,就是不好(敢)想,思维卡住了