9.24 达哥
思路:
计数四合一,考察组合数,卡特兰数,动态规划.
对于n<=100的40%数据:存在一个通用的DP,定义f[i][j][k]表示i步之后走到(j,k)的方案数,复杂度为O(n^3).
对于typ=1的数据:答案为catalan数,使用O(n)的catalan数递推公式或者利用组合数O(1)计算均可.catalan(n)=C(2n,n)/(n+1)
对于typ=0的数据:枚举横向移动了多少步.横向移动i步时(为了存在合法解,i必须是偶数),方案数为C(n,i) * C(i,i/2) * C((n-i),(n-i)/2)
对于typ=2的数据:f[i]表示走了i步回到原点的方案数,枚举第一次回到原点时走过的步数j(为了存在合法解,j为偶数),则此时方案数为f[i-j]*catalan(j/2-1),复杂度为O(n^2)所以最大范围只出到1000.
为什么这里是j/2-1呢,因为我们要保证这是第一次走回来,所以在走这j步的过程中是不能够走回来的,相当于是一个括号匹配,要求保证中途不能有完整匹配,这么做到这一点呢,只需要拿出一组前后括号,前括号放开头,后括号放末尾,期间再求卡特兰数就不会统计到完美匹配了。所以只有j/2-1组括号。
对于typ=3的数据:枚举横向移动了多少步.横向移动i步时(为了存在合法解,i必须是偶数),方案数为C(n,i) * catalan(i/2) * catalan((n-i)/2)
n<=1000的数据是为了照顾只会使用杨辉三角形求解组合数的选手.
#include <cstdio>
#include <iostream>
#define maxn 200005
#define mod 1000000007
using namespace std;
int mub[maxn], inv[maxn];
int C(int n,int m){
if(n < m) return 0;
return mub[n] * 1ll * inv[m] % mod * inv[n-m] % mod;
}
int qpow(int a, int b){
int res = 1;
for( ; b; b>>=1, a=a*1ll*a % mod){
if(b & 1) res = res*1ll*a % mod;
}
return res;
}
int catalan(int n){
return C(n*2, n) * 1ll * qpow(n+1, mod-2) % mod;
}
void init(){
mub[0] = 1;
for(int i=1; i<maxn; ++i) mub[i] = mub[i-1] * 1ll * i % mod;
inv[maxn-1] = qpow(mub[maxn-1], mod-2);
for(int i=maxn-1; i>=1; --i) inv[i-1] = inv[i] * 1ll * i % mod;
}
int f[maxn];
int main(){
freopen("problem.in","r",stdin);
freopen("problem.out","w",stdout);
init();
int n, opt; scanf("%d%d", &n, &opt);
if(opt == 0){
int ans = 0;
for(int i=0; i<=n; i+=2){
ans = (ans + C(n,i) * 1ll * C(i,i/2) % mod * C((n-i),(n-i)/2) % mod) % mod;
}//先选i步左右走,在i步中选左右,再从剩下的n-i步中选上下
printf("%d\n", ans);
}
else if(opt == 1){
printf("%d\n", catalan(n/2));
}//类似括号匹配
else if(opt == 2){
n /= 2; f[0] = 1; f[1] = 4;//f[i]表示走了2*i步之后到原点的方案数
for(int i=2; i<=n; ++i){
for(int j=1; j<=i; ++j){
f[i] = (f[i] + f[i-j] * 4ll * catalan(j-1) % mod) % mod;
}//第一次回到原点时走过2*j步
}
printf("%d\n",f[n]);
}
else{
int ans = 0;
for(int i=0; i<=n; i+=2){
ans = (ans + C(n,i) * 1ll * catalan(i/2) % mod * catalan((n-i)/2) % mod) % mod;
}//两次括号匹配
printf("%d\n", ans);
}
return 0;
}