Codeforces1548 C. The Three Little Pigs（组合数学+dp）

题意：

解法：

$$\begin{gathered} 对于给定的x,ans=\sum _{i=0}^n{C}_{3i}^x \\ C(n,m)可以预处理逆元然后O(1)算. \\ 对于单个x,累加C(3i,x)可以O(n)计算答案. \\ 但是这题是多组数据,需要优化. \\ 考虑不同的x之间是否存在递推关系. \\ 即考虑利用x=t的答案推出x=t+1的答案. \\ ----- \\ 已知\sum _{i=0}^n{C}_{3i}^x,要推出\sum _{i=0}^n{C}_{3i}^{x+1} \\ 令d[x][j=0/1/2]表示\sum _{i=0}^n{C}_{3i+j}^x \\ 那么有式子: \\ 1.d[x+1][1]=d[x+1][0]+d[x][0] \\ 2.d[x+1][2]=d[x+1][1]+d[x][1] \\ 3.d[x+1][0]+d[x+1][1]+d[x+1][2]=C_{3n+3}^{x+2} \\ 式子解释: \\ 1和2: \\ d[x+1][1]=\sum _{i=0}^n{C}_{3i+1}^{x+1} \\ =\sum _{i=0}^n[C_{3i}^{x+1}+C_{3i}^x] \\ =\sum _{i=0}^n{C}_{3i}^{x+1}+\sum _{i=0}^n{C}_{3i}^x \\ =d[x+1][0]+d[x][0] \\ d[x+1][2]同理. \\ 3: \\ d[x+1][0]+d[x+1][1]+d[x+1][2] \\ =\sum _{i=0}^n[C_{3i}^{x+1}+C_{3i+1}^{x+1}+C_{3i+2}^{x+1}] \\ =\sum _{i=0}^{3n+2}{C}_i^{x+1}=C_{3n+3}^{x+2} \\ 那么d[x+1][0]=C_{3n+3}^{x+2}-d[x+1][1]-d[x+1][2] \\ 三个变量三个式子,可以推出转移方程: \\ 1.d[x+1][0]=(C_{3\ast n+3}^{x+2}-2d[x][0]-d[x][1])/3 \\ 2.d[x+1][1]=d[x+1][0]+d[x][0] \\ 3.d[x+1][2]=d[x+1][1]+d[x][1] \\ 边界: \\ d[0][0]=d[0][1]=d[0][1]=n+1 \\ 答案: \\ 对于给定的x,d[x][0]就是答案. \end{gathered}$$

code：

#include<bits/stdc++.h>
// #define MULTI_CASE
// #define SYNC_OFF
#define PI pair<int,int>
// #define int long long
using namespace std;
// const int mod=998244353;
const int mod=1e9+7;
const int maxm=6e6+5;
int n,q;
int d[maxm][3];
int fac[maxm],inv[maxm];
int ppow(int a,int b,int mod){
    int ans=1%mod;a%=mod;
    while(b){if(b&1)ans=1ll*ans*a%mod;a=1ll*a*a%mod;b>>=1;}
    return ans;
}
int C(int n,int m){
    if(m<0||m>n)return 0;
    return 1ll*fac[n]*inv[m]%mod*inv[n-m]%mod;
}
void C_init(){
    fac[0]=1;
    for(int i=1;i<maxm;i++)fac[i]=1ll*fac[i-1]*i%mod;
    inv[maxm-1]=ppow(fac[maxm-1],mod-2,mod);
    for(int i=maxm-2;i>=0;i--)inv[i]=1ll*inv[i+1]*(i+1)%mod;
}
void solve(){
    C_init();
    scanf("%d%d",&n,&q);
    d[0][0]=d[0][1]=d[0][2]=n+1;
    int inv3=ppow(3,mod-2,mod);
    for(int x=0;x<n*3;x++){
        d[x+1][0]=(1ll*C(3*n+3,x+2)-d[x][1]-2*d[x][0])%mod*inv3%mod;
        d[x+1][1]=(d[x+1][0]+d[x][0])%mod;
        d[x+1][2]=(d[x+1][1]+d[x][1])%mod;
    }
    for(int i=1;i<=q;i++){
        int x;scanf("%d",&x);
        int ans=d[x][0];
        ans=(ans%mod+mod)%mod;
        printf("%d\n",ans);
    }
}
void Main(){
    #ifdef MULTI_CASE
    int T;cin>>T;while(T--)
    #endif
    solve();
}
void Init(){
    #ifdef SYNC_OFF
    ios::sync_with_stdio(0);cin.tie(0);
    #endif
    #ifndef ONLINE_JUDGE
    freopen("../in.txt","r",stdin);
    freopen("../out.txt","w",stdout);
    #endif
}
signed main(){
    Init();
    Main();
    return 0;
}

---