题目链接:吴老板教课 (nowcoder.com)
题目描述
无所不能的吴老板经常教导他学生,学数学要临场快速突击:
那么现在吴老板给大家一个数论公式
令h(0)=1,h(1)=1,
卡特兰数满足递推式 :
h(n)= h(0)*h(n-1)+h(1)*h(n-2) + ... + h(n-1)*h(0) (n>=2)
通项公式可以写成:
h(n)= C(n, 2n) - C(n +1, 2n) (C是组合数符号)
吴老板给大家的任务很简单,结合以上公式,求出第1项到第n项的卡特兰数的和(结果对100000007取模)
输入描述:
输入一个整数n
输出描述:
输出一个整数代表前n个卡特兰数的和
示例1
输入
复制
3
输出
复制
8
说明
1+2+5=8
备注:
2<=n<=100000 (注意:如你不会逆元,那你就需要自己推出卡特兰数另一个递推公式了)
思路:
先普及一下什么是卡特兰数:给出一个n*n的正方形棋盘,求从左下角到右上角的不经过对角线的所有走法刚好是F[n],即C(2n, n)-C(2n, n-1)
就是卡特兰数前n项和。
AC代码:
#include<bits/stdc++.h>
using namespace std;
#define LL long long
const LL N=1e6+10;
LL f[N],rf[N],sum[N];
const LL mod=1e8+7;
LL n;
inline LL ksm(LL a, int p)//快速幂
{
a = a % mod;
LL res = 1;
while(p)
{
if(p&1) res = res * a % mod;
a = a * a % mod;
p >>= 1;
}
return res;
}
inline LL C(int a,int b)
{
return f[a] * rf[b] % mod * rf[a-b] % mod;
}
inline void init()
{
f[0] = 1;
for(int i=1; i<=n; i++) //预处理出阶乘
f[i] = f[i-1] * i % mod;
rf[n] = ksm(f[n], mod - 2);
for (int i = n - 1; i >= 0; i -- )//阶乘逆元
rf[i] = rf[i + 1] * (i + 1) % mod;
}
signed main(){
// LL sum=0;
ios;
cin>>n;
int m=n;
n*=2;
init();
for(int i=2;i<=m;i++){
sum[i]=(mod+C(i*2,i)-C(i*2,i+1))%mod;
sum[i]+=sum[i-1];
sum[i]%=mod;
}
cout<<sum[m]+1<<"\n";
}