【NOIP模拟赛】长寿花（组合计数，DP去重）

$n,m<=1e6,a_i<=5e3,\sum a_i<=1e6$
$\text{不}用谢。$

首先每一层的答案只与颜色的数量有关。
但是相邻层的颜色集合不能相同。
预处理$g_{i,j}$表示$i$个位置分配给$j$种颜色并且每种都出现并且相邻的颜色不同的方案数，注意我们让$g$中统计的方案颜色无标号之分，也就是$g_{i,j}=g_{i-1,j-1}+g_{i-1,j}\ast (j-1)$，这样可以在之后的$\mathrm{D}\mathrm{P}$中可以免去求逆元的环节。
$f_{i,j}$表示前$i$层，第$i$层用了$j$种颜色的方案。
$f_{i,j}=g_{a_i,j}\left(\genfrac{}{}{0ex}{}{m}{j}\right)j!{\sum }_k{f}_{i-1,k}-f_{i-1,j}\ast g_{i-1,j}j!$
这里的$\left(\genfrac{}{}{0ex}{}{m}{j}\right)j!$可以简单预处理递推得到而不用求逆元。

$\mathrm{A}\mathrm{C}\mathrm{C}\mathrm{o}\mathrm{d}\mathrm{e}$

#include<bits/stdc++.h>
#define maxn 10000005
using namespace std;
 
int n,m,p;
int a[maxn],g[5005][5005],f[2][5005],Am[5005],fac[5005];
 
int main(){
//  freopen("1.in","r",stdin);
//  freopen("1.out","w",stdout);
    scanf("%d%d%d",&n,&m,&p);
    g[0][0] = 1;
    for(int i=1;i<=5000;i++)
        for(int j=1;j<=i && j<=m;j++)
            g[i][j] = ((j ? g[i-1][j-1] : 0) + 1ll * g[i-1][j] * (j-1)) % p;
    Am[0] = fac[0] = 1;
    for(int i=1;i<=min(5000,m);i++) Am[i] = 1ll * Am[i-1] * (m-i+1) % p , fac[i] = 1ll * fac[i-1] * i % p;
    int now = 1 , pre = 0 , sum = 0;f[now][0] = sum = 1;
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        swap(now,pre);
        for(int j=0;j<=a[i];j++)
            f[now][j] = sum * 1ll * g[a[i]][j] % p * Am[j] % p; 
        for(int j=0;j<=a[i-1];j++) if(f[pre][j]){
            f[now][j] = (f[now][j] - 1ll * g[a[i]][j] * f[pre][j] % p * fac[j]) % p;
            f[pre][j] = 0;
        }
        sum = 0;
        for(int j=0;j<=a[i];j++) sum = (sum + f[now][j]) % p /* , printf("%d %d %d\n",i,j,f[now][j])*/;
    }
    printf("%d\n",(sum+p)%p);
}