BZOJ4313 三维积木 计数 dp

题目链接

题意：
你有r块红色积木，b块蓝色积木和g块绿色积木，积木都是
$1\ast 1\ast 1$的立方体，你要在n*n的格子上摆出主视图积木颜色全部相同的三维物体，要求所有的积木都要用上，问方案数，对1e9+7取模。r,b,g,n<=25

题解：
一看就是一个比较复杂的计数题，一般还是从dp入手。我们先假设三种颜色中的某一种为主视图的颜色，那么对于其他两种颜色，我们可以先求出不对其进行区分的方案，再乘一个组合数$C_{两种颜色总数}^{其中一种颜色的数量}$就是区分后的答案。

我们先考虑对于一个1行n列，高度不限的空间放这么多积木的方案数。我们设$f[i][j][k][p][q]$表示填了前$i$列，当前列放了$j$块积木，目前填的所有列中最高的一列的高度是$k$，当前列放了$p$块主视图颜色的积木和$q$块非主视图颜色的积木的方案数。我们可以分类转移，如果$j=k$那么如果我们在当前这一列在加一个和主视图颜色一样的积木，那么就可以转移到最高的一列高度更高的状态。否则只能转移到最高一列仍然是原来最高高度的状态，然后枚举是加了一个主视图颜色的还是非主视图颜色的积木就行了，方程很简单，需要的话看看代码就好了。这一步复杂度是$O(n^5)$的，空间复杂度的话可以滚动数组滚掉第一维，是$n^4$的。

然后我们可以通过刚才的答案，计算出在1行n列，高度不限的情况下，用$x$个主视图颜色的积木和$y$个非主视图颜色的积木有多少种不同的方案，我们记作$g[x][y]$，我们可以枚举最高的高度，然后枚举$x$和$y$，这样$n^3$求出所有的$g[x][y]$的答案。

有了一行多列的$g[x][y]$的答案，我们就可以求多行多列的答案了。我们设$dp[i][j][k]$为前$i$行，放了$j$个主视图颜色的积木和$k$个非主视图颜色的积木的方案数。我们发现行与行之间互不干扰，满足乘法原理，于是我们可以枚举$i,j,k$，再枚举当前行放$p$个主视图颜色的积木和$q$个非主视图颜色的积木，我们设三种颜色中数量最多的木块的数量是$m$，那么转移方程式是$dp[i+1][j+p][k+q]={\sum }_{i=0}^n{\sum }_{j=0}^m{\sum }_{k=0}^{2m}{\sum }_{p+j<=m}{\sum }_{q+k<=2m}dp[i][j][k]\ast g[p][q]$

最后统计答案的时候只需要枚举一下主视图颜色是哪一个，然后求出来的是不区分另外两种颜色的结果，所以再乘一个一开始时说的组合数，就是最终的答案了。

代码：

#include <bits/stdc++.h>
using namespace std;

int r,g,b,n,opt,m;
const long long mod=1e9+7;
long long f[2][27][27][27][55],gg[55][55],dp[27][55][55],c[55][55],ans; 
int main()
{
 scanf("%d%d%d%d",&r,&g,&b,&n);
 m=max(r,max(g,b));
 f[0][0][0][0][0]=1;
 for(int i=0;i<n;++i)
 {
  for(int j=0;j<=m;++j)
  {
   for(int k=j;k<=m;++k)
   {
    for(int p=k;p<=m;++p)
    {
     for(int q=0;q<=2*m;++q)
     f[opt^1][j][k][p][q]=0;
    }
   }
  }
  for(int j=0;j<=m;++j)
  {
   for(int k=j;k<=m;++k)
   {
    for(int p=k;p<=m;++p)
    {
     for(int q=0;q<=2*m;++q)
     {
      if(f[opt][j][k][p][q])
      {
       f[opt^1][0][k][p][q]+=f[opt][j][k][p][q];
       if(f[opt^1][0][k][p][q]>=mod)
       f[opt^1][0][k][p][q]-=mod;
       if(j==k)
       {
        f[opt][j+1][k+1][p+1][q]+=f[opt][j][k][p][q];
        if(f[opt][j+1][k+1][p+1][q]>=mod)
        f[opt][j+1][k+1][p+1][q]-=mod;
       }
       else
       {
        f[opt][j+1][k][p+1][q]+=f[opt][j][k][p][q];
        if(f[opt][j+1][k][p+1][q]>=mod)
        f[opt][j+1][k][p+1][q]-=mod;
        f[opt][j+1][k][p][q+1]+=f[opt][j][k][p][q];
        if(f[opt][j+1][k][p][q+1]>=mod)
        f[opt][j+1][k][p][q+1]-=mod;
       }
      }
     }
    }
   }
  }
  opt^=1;
 }
 for(int i=0;i<=m;++i)
 c[i][0]=1;
 for(int i=1;i<=m*2;++i)
 {
  for(int j=1;j<=i;++j)
  {
   c[i][j]=c[i-1][j-1]+c[i-1][j];
   if(c[i][j]>=mod)
   c[i][j]-=mod;
  }
 }
 for(int i=0;i<=m;++i)
 {
  for(int j=i;j<=m;++j)
  {
   for(int k=0;k<=2*m;++k)
   {
    if(f[opt][0][i][j][k])
    {
     gg[j][k]+=f[opt][0][i][j][k];
     if(gg[j][k]>=mod)
     gg[j][k]-=mod;
    }
   }
  }
 }
 dp[0][0][0]=1;
 for(int i=0;i<n;++i)
 {
  for(int j=0;j<=m;++j)
  {
   for(int k=0;k<=2*m;++k)
   {
    if(dp[i][j][k])
    {
     for(int p=0;p+j<=m;++p)
     {
      for(int q=0;q+k<=2*m;++q)
      {
       dp[i+1][p+j][q+k]+=dp[i][j][k]*gg[p][q]%mod;
       if(dp[i+1][p+j][q+k]>=mod)
       dp[i+1][p+j][q+k]-=mod;
      }
     }
    }
   }
  }
 }
 ans=dp[n][r][g+b]*c[g+b][b]%mod;
 ans+=dp[n][g][r+b]*c[r+b][b]%mod;
 if(ans>=mod)
 ans-=mod;
 ans+=dp[n][b][r+g]*c[r+g][g]%mod;
 if(ans>=mod)
 ans-=mod;
 printf("%lld\n",ans);
 return 0;
}