题目描述
铁路进行列车调度时,常把站台设计成栈式结构的站台,试问: 设有编号为1到n的n辆列车,顺序开入栈式结构的站台,则可能的出栈序列有多少种?
输入
输入包含多组测试数据。每组为一个正整数n(1<=n<=20),表示有n辆列车。
输出
输出可能的出栈序列有多少种。
样例输入
4
3
样例输出
14
5
思路:数据结构中很简单的问题。卡特兰数问题。有n个数字依次入栈,则所有合法的出栈数为C(2n,n) / (n+1)
一开始写出了代码,但是只过了一半的样例。后来才知道由于某些原因,卡特兰数公式需要先化简才能使用。
下面是最开始写的代码:
#include <iostream>
using namespace std;
typedef long long ll;
ll fac(int x)
{
//求阶乘
ll ans = 1;
for(int i = 1;i <= x;i++)
{
ans *= i;
}
return ans;
}
int main()
{
int n;
while(cin >> n)
{
//res = 2n!/n!/n!/(n+1) = 2n!/n!/(n+1)!
ll res = fac(2 *n) / fac(n) / fac(n + 1);
cout << res << endl;
}
return 0;
}
C(2n,n)
= ( 2n)!/ n!/(2n - n)!
= (2n)!/ n!/ n!
= ( 1 * 2 * 3 * 4 * 5 * 6 * … * n * n+1 * n+2 * …* 2n) / (1 * 2 * 3 * 4 * 5 * 6 * … * n) / (1 * 2 * 3 * 4 * 5 * 6 * … * n)
= (n+1 * n+2 * …* 2n ) / (1 * 2 * 3 * 4 * 5 * 6 * … * n )
可发现,C(2n,n)的分母从1乘到n,故可用for循环实现,用 i 枚举1 ~ n,每次分母都除以 i 。同理,分子从 n + 1乘到 n + n,也可以用for循环来实现,用 i 枚举1 ~ n,每次都乘以n + i即可
下面是本题AC代码
#include <iostream>
#include <cstdio>
using namespace std;
int main()
{
int n;
while (cin >> n)
{
long long ans = 1;
//卡特兰数化简结果
for (int i = 1;i <= n;i++)
{
//每次分子都乘以n + i
//每次分母都除以 i
ans = ans * (n + i) / i;
}
//最终得结果:C(2n,n)/(n+1)
cout << ans / (n + 1) << endl;
}
return 0;
}