原题链接:
题解:
本题所要求的是卡特兰数。在本题中,我们可以将长度为2n的01序列视为从(0,0)到坐标(n,n)的路径,0代表右移,1代表上移。那么从(0,0)到坐标(n,n)的合法路径即为我们要求的满足条件的01序列(也即当下情景的卡特兰数)
目标:求从(0,0)到坐标(n,n)的合法路径数量。
由于我们目前不能直观的计算从(0,0)到坐标(n,n)的合法路径数量,因此可以采用间接计算方式:
从(0,0)到坐标(n,n)的合法路径数量=从(0,0)到坐标(n,n)的路径总数量-从(0,0)到坐标(n,n)的非法路径数量
从(0,0)到坐标(n,n)的路径总数量=,这是因为(如果我们考虑一个 n×n 的网格坐标系,从左下角移动到右上角,每次只能向上或向右移动一步。这是一个典型的组合问题,其中每次移动都可以看作是在一个二叉树中向左或向右移动。而总共需要进行 2n 次移动,其中 n 次是向上移动,n 次是向右移动。所以,总的移动方式数量为选择 n 次向上移动的方式,即 C(2n,n)。
那么非法路径数量如何求呢?我们可以直观发现只要路径接触下图红线部分(包括红线)就是非法的,而从(0,0)到坐标(n,n)的路径则关于红线与其从(0,0)到坐标(n-1,n+1)的路径对称,而从(0,0)到坐标(n-1,n+1)的路径都是非法的(因为接触到了红线),那么从(0,0)到坐标(n,n)的非法路径数量=(0,0)到坐标(n-1,n+1)的路径总数量=.
因此卡特兰数就等于:
详细的推导证明过程:
参见:AcWing 889. 满足条件的01序列 - AcWing
代码:
//卡特兰数:满足条件的01序列
#include <bits/stdc++.h>
using namespace std;
using LL = long long;
const int mod = 1e9 + 7;
int qmi(int a, int k, int p) {
int res = 1 % p;
while (k) {
if (k & 1) res = (LL)res * a % p;
a = (LL)a * a % p;
k >>= 1;
}
return res;
}
int main() {
int n;
cin >> n;
int a = 2 * n, b = n;
int res = 1;
for (int i = a; i > a - b; i--) res = (LL)res * i % mod;
for (int i = 1; i <= b; i++) res = (LL)res * qmi(i, mod - 2, mod) % mod;
res = (LL)res * qmi(n + 1, mod - 2, mod) % mod;
cout << res << endl;
}