CF1548C The Three Little Pigs(组合数学+dp)

题目传送门

题意

这道题实际上就是让求$\sum _{i=1}^n{C}_{3\ast i}^x,x$是我们每次询问的数值，$n$是小猪进场的轮数。

思路

我们可以定义一个$dp$方程，$dp[x][m]$代表$\sum _{i=1}^{n-1}{C}_{3\ast i+m}^x$
因此，我们可以得到$dp[x][0]+dp[x][1]+dp[x][2]=\sum _{i=1}^{3\ast n-1}{C}_i^x$
又根据杨辉三角的性质我们可以得到

$$C_i^x+C_i^{x-1}=C_{i+1}^x$$

因此

$dp[x][1]=dp[x][0]+dp[x-1][0]$
$dp[x][2]=dp[x][1]+dp[x-1][1]$

又根据公式
$$\sum _{i=k}^n{C}_i^k=C_{n+1}^{k+1}$$
因此$dp[x][0]+dp[x][1]+dp[x][2]=C_{3\ast n}^{x+1}$

现在递推式每一个状态下的$dp[x][1],dp[x][2]$都能求出来了，现在就来求$dp[x][0]$
根据上述方程。

$dp[x][1]=dp[x][0]+dp[x-1][0]$
$dp[x][2]=dp[x][1]+dp[x-1][1]$
$dp[x][0]+dp[x][1]+dp[x][2]=C_{3\ast n}^{x+1}$

代入原方程求得：

$dp[x][0]+dp[x][0]+dp[x-1][0]+dp[x][1]+dp[x-1][1]=C_{3\ast n}^{x+1}$
$dp[x][0]+dp[x][0]+dp[x-1][0]+dp[x][0]+dp[x-1][0]+dp[x-1][1]=C_{3\ast n}^{x+1}$

整理一下：

$3\ast dp[x][0]+2\ast dp[x-1][0]+dp[x-1][1]=C_{3\ast n}^{x+1}$

因此

$dp[x][0]=(C_{3\ast n}^{x+1}-2\ast dp[x-1][0]-dp[x-1][1])/3$

同时，官方题解的逆元求组合数的板子比我的强了不知道多少倍。

#include<vector>
#define vi vector<int>
#define vl vector<long long>
typedef long long ll;
const ll MOD=1e9+7;
struct Combo {
    vl facs;
    vl invfacs;
    int N;
 
    Combo(int N) {
        this->N=N;
        facs.assign(N+1,0);
        invfacs.assign(N+1,0);
        facs[0] = 1;
        for (int i = 1; i <= N; i++) {
            facs[i] = (facs[i-1]*i)%MOD;
        }
        invfacs[N] = power(facs[N],MOD-2);
        for (int i = N-1; i >= 0; i--) {
            invfacs[i] = (invfacs[i+1]*(i+1))%MOD;
        }
    }
 
    ll choose(int n, int k) {
        if (n<0||k<0||n<k) return 0LL;
        ll denInv = (invfacs[k]*invfacs[n-k])%MOD;
        ll ans = (facs[n]*denInv)%MOD;
        return ans;
    }
 
    ll power(ll x, ll y) {
        ll ans = 1;
        x %= MOD;
        while (y > 0) {
            if (y%2==1) ans = (ans*x)%MOD;
            y /= 2;
            x = (x*x)%MOD;
        }
        return ans;
    }
};

#include<iostream>
#include<vector>
#define vi vector<int>
#define vl vector<long long>
using namespace std;
typedef long long ll;
const ll N=1e9+7;
const int M=3e6+5;
const ll MOD=1e9+7;
struct Combo {
    vl facs;
    vl invfacs;
    int N;
 
    Combo(int N) {
        this->N=N;
        facs.assign(N+1,0);
        invfacs.assign(N+1,0);
        facs[0] = 1;
        for (int i = 1; i <= N; i++) {
            facs[i] = (facs[i-1]*i)%MOD;
        }
        invfacs[N] = power(facs[N],MOD-2);
        for (int i = N-1; i >= 0; i--) {
            invfacs[i] = (invfacs[i+1]*(i+1))%MOD;
        }
    }
 
    ll choose(int n, int k) {
        if (n<0||k<0||n<k) return 0LL;
        ll denInv = (invfacs[k]*invfacs[n-k])%MOD;
        ll ans = (facs[n]*denInv)%MOD;
        return ans;
    }
 
    ll power(ll x, ll y) {
        ll ans = 1;
        x %= MOD;
        while (y > 0) {
            if (y%2==1) ans = (ans*x)%MOD;
            y /= 2;
            x = (x*x)%MOD;
        }
        return ans;
    }
};
ll dp[M][4];
int main(){
	ll n;
    cin>>n;
	Combo c(3*n);
    ll inv=c.power(3,MOD-2);
    dp[0][0]=dp[0][1]=dp[0][2]=n;
    for(int i=1;i<n*3;i++){
    	dp[i][0]=(c.choose(3*n,i+1)-2*dp[i-1][0]-dp[i-1][1]);if(dp[i][0]<=0)dp[i][0]+=((-dp[i][0])/MOD+1)*MOD; 
		dp[i][0]=dp[i][0]*inv%MOD;
    	dp[i][1]=(dp[i-1][0]+dp[i][0])%MOD;
    	dp[i][2]=(dp[i-1][1]+dp[i][1])%MOD;
	}
    int m;
    cin>>m;
    while(m--){
    	int q;
    	scanf("%d",&q);
    	printf("%lld\n",(ll)(dp[q][0]+c.choose(3*n,q))%MOD);
	}
}