CodeForces-140 E New Year Garland(DPカウント)

に言及
m色のカラーボールでn階のクリスマスツリーを飾る.クリスマスツリーのi番目のレイヤは、l[i]個のカラーボールが1列に並んでおり、同じレイヤ内の隣接カラーボールの色が異なり、隣接する2つのレイヤで使用されるカラーボールの色の集合が異なる.どれだけの種類の装点の方案があることを求めて、答えはpに対して型を取ります.
タイトルリンク
問題解
数日待って書く
コード#コード#

#include 
using namespace std;
typedef long long ll;

int mod,m,n;
ll xx[5555],c[5555],a[5555][5555];
ll dp[2][5555];

void init(int m)
{
    a[0][0]=1;
    for(int i=1;i<=5000;++i)
        for(int j=1;j<=min(m,i);++j)
            a[i][j]=(a[i-1][j]*(j-1)%mod+a[i-1][j-1]%mod)%mod;//          
    xx[1]=m;
    for(int i=2;i<=min(m,5000);++i)
        xx[i]=(xx[i-1]*(m-i+1))%mod;//     
    c[1]=1;
    for(int i=2;i<=min(m,5000);++i)
        c[i]=(c[i-1]*i)%mod;//    
}

int main()
{
    scanf("%d%d%d",&n,&m,&mod);
    init(m);
    int l,pos=0;
    ll sum,tmp;
    for(int s=0;ss)
    {
        scanf("%d",&l);
        tmp=0;
        memset(dp[pos^1],0,sizeof(dp[pos^1]));
        if(s==0)//   
        {
            for(int i=1;i<=min(m,l);++i)
            {
                dp[pos^1][i]=(xx[i]*a[l][i])%mod;
                tmp=(tmp+dp[pos^1][i])%mod;
            }
        }
        else
        {
            for(int i=1;i<=min(m,l);++i)
            {
                dp[pos^1][i]=(((((((xx[i]*sum)%mod)-((c[i]*dp[pos][i])%mod))%mod)+mod)%mod)*a[l][i])%mod;
                tmp=(tmp+dp[pos^1][i])%mod;
            }
        }
        sum=tmp;
        pos^=1;//      i i+1  ，       ，    
    }
    printf("%lld
",sum);
    return 0;
}