思路
改为所有 ,都有 。
观察发现,第 行 ,第 行能取 的位置是 和 的一个前缀。形如:
000010110
000010100
010110100
000000000
100000000
但可以空一些行和列,不方便,考虑将所有有 的行和列压起来。设 表示前 行,有 个列有过 ,当前行选 个,强制连续选。首先可以取一个前缀,,后缀和维护。其次可以向前在 任意取,但强制连续选,枚举选 个,,维护一个斜线的前缀和。
计算答案,对于每个 , 行选 行, 列选 列,其他放 ,。再加上全取 的情况。
注意取模优化和枚举时 的常数。
code
for(int i=1;i<=m;i++)dp[i][i]=1,ans=add(ans,C(m,i)*C(n,1)%mod);
for(int i=2;i<=n;i++){
for(int j=1;j<=m;j++){
for(int k=j;~k;k--)f[j][k]=add(f[j][k+1],dp[j][k]);
}
for(int j=1;j<=m;j++){
for(int k=1;k<=j;k++)f[j][k]=add(f[j][k],f[j-1][k-1]);
}
for(int j=1;j<=m;j++){
for(int k=1;k<=j;k++){
dp[j][k]=f[j][k];
}
}
for(int j=1;j<=m;j++){
for(int k=1;k<=j;k++){
ans=add(ans,C(m,j)*C(n,i)%mod*dp[j][k]%mod);
}
}
}
printf("%lld\n",ans+1);