题目背景
栈是计算机中经典的数据结构,简单的说,栈就是限制在一端进行插入删除操作的线性表。
栈有两种最重要的操作,即 pop(从栈顶弹出一个元素)和 push(将一个元素进栈)。
栈的重要性不言自明,任何一门数据结构的课程都会介绍栈。宁宁同学在复习栈的基本概念时,想到了一个书上没有讲过的问题,而他自己无法给出答案,所以需要你的帮忙。
题目描述
宁宁考虑的是这样一个问题:一个操作数序列, 1 , 2 , … , n 1,2,\ldots ,n 1,2,…,n(图示为 1 到 3 的情况),栈 A 的深度大于 n n n。
现在可以进行两种操作,
- 将一个数,从操作数序列的头端移到栈的头端(对应数据结构栈的 push 操作)
- 将一个数,从栈的头端移到输出序列的尾端(对应数据结构栈的 pop 操作)
使用这两种操作,由一个操作数序列就可以得到一系列的输出序列,下图所示为由 1 2 3
生成序列 2 3 1
的过程。
(原始状态如上图所示)
你的程序将对给定的 n n n,计算并输出由操作数序列 1 , 2 , … , n 1,2,\ldots,n 1,2,…,n 经过操作可能得到的输出序列的总数。
输入格式
输入文件只含一个整数 n n n( 1 ≤ n ≤ 18 1 \leq n \leq 18 1≤n≤18)。
输出格式
输出文件只有一行,即可能输出序列的总数目。
样例 #1
样例输入 #1
3
样例输出 #1
5
提示
【题目来源】
NOIP 2003 普及组第三题
思路:这道题我用递归的思路来做,首先想到的方法是dp,即动态规划:
dp[i,j]表示i个输出序列中的数,j个出栈的所有可能数。
动态规划的初始条件:
定义dp[0,j] = 1,意思是输出序列中的数为0,只有出栈的可能为1种,不难想到,从栈中取出,无论多少个,结果都只有一种可能。
递归条件:
dp[i,j] = dp[i-1,j] + dp[i,j-1] 意思是输出序列中有i个数,j个出栈的可能数也就等于输出序列中有i-1个数,j个出栈的可能数加上输出序列中有i-1个数,j个出栈的可能数
特别的:
dp[i,j]当i = j时,说明栈空了,此时dp[i,j] = dp[i-1,j]
于是就有了下面的python代码
n = int(input())
dp = [[0 for i in range(n + 1)] for j in range(n + 1)]
for i in range(n + 1):
dp[0][i] = 1
for i in range(1, n + 1, 1):
for j in range(1, n + 1, 1):
if i == j:
dp[i][j] = dp[i - 1][j]
else:
dp[i][j] = dp[i][j - 1] + dp[i - 1][j]
print(dp[n][n])