统计和生成所有不同的二叉树(进阶)

统计和生成所有不同的二叉树(进阶)

题目描述

给出一个整数 n，如果 n < 1，代表空树，否则代表中序遍历的结果为 {1, 2, 3… n}。请输出可能的二叉树结构有多少。

输入描述：

第一行输入一个整数 n。

输出描述：

输出一个整数对 1e9 + 7 取模的值表示答案。

示例1

输入

8

输出

1430

备注：

$1\le n\le 10^6$

---

题解：

数据范围意味着我们不能采用前面 $O(n^2)$ 的 dp 解法了，打印结果可以发现结果是卡特兰数：$\frac{\left(\genfrac{}{}{0ex}{}{2n}{n}\right)}{n+1}$。

但是此题涉及到除法取模问题，我们可以将其转换为乘法，即：$(a/b)\%M=(a\ast b^{-1})\%M$ ，其中 $b^{-1}$ 表示 b 的逆元。求逆元可以用逆元公式：$inv[i]=(M-M/i)\ast inv[M\%i]\%M$。同样求逆元还有其它公式，请参考逆元，讲解很详细。

代码：

#include <cstdio>
using namespace std;

const int N = 1000010;
const int MOD = 1e9 + 7;

typedef long long LL;

int n;
LL inv[N];

LL calc() {
    inv[0] = inv[1] = 1;
    for (int i = 2; i <= n + 1; ++i)
        inv[i] = (MOD - MOD / i) * inv[MOD % i] % MOD;
    int m = n << 1;
    LL ret = 1;
    for (int i = 0; i < n; ++i)
        ret = ret * (m - i) % MOD * inv[i + 1] % MOD;
    return ret * inv[n + 1] % MOD;
}

int main(void) {
    scanf("%d", &n);
    printf("%lld\n", calc());
    return 0;
}