题(problem) （组合求方案数）

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;
}