火车出站(卡特兰数)

题目描述

铁路进行列车调度时，常把站台设计成栈式结构的站台，试问： 设有编号为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;
}