题目
有n个木块排成一行,从左到右依次编号为1~n。你有k种颜色的油漆,其中第i种颜色的油漆足够涂ci个木块。
所有油漆刚好足够涂满所有木块,即c1+c2+...+ck=n。相邻两个木块涂相同色显得很难看,所以你希望统计任意两
个相邻木块颜色不同的着色方案。(100%的数据满足:1 <= k <= 15, 1 <= ci <= 5)
题解
计数类DP
看到ci如此小,不妨从ci下手。设f[c5][c4][c3][c2][c1][last]表示有5次使用机会的有c5个,……,1次的有c1个,最后一次使用的是剩余使用次数last的颜料。
转移时,需要注意这次使用的颜料l,下次使用时不能再使用颜料l。所以次数要减1。方案数=使用5次+使用4次……
代码
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=20;
const int mod=1e9+7;
ll f[16][16][16][16][16][16];
bool v[16][16][16][16][16][6];
ll dfs(int a,int b,int c,int d,int e,int l)
{
if(v[a][b][c][d][e][l]) return f[a][b][c][d][e][l];
v[a][b][c][d][e][l]=true;
ll re=0;
if(a>0) re+=dfs(a-1,b+1,c,d,e,5)*a;
if(b>0) re+=dfs(a,b-1,c+1,d,e,4)*(b-(l==5));
if(c>0) re+=dfs(a,b,c-1,d+1,e,3)*(c-(l==4));
if(d>0) re+=dfs(a,b,c,d-1,e+1,2)*(d-(l==3));
if(e>0) re+=dfs(a,b,c,d,e-1,0) *(e-(l==2));
return f[a][b][c][d][e][l]=re%mod;
}
int s[maxn];
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
int col;
scanf("%d",&col);
s[col]++;
}
// memset(f,-1,sizeof(f));
for(int i=0;i<=5;i++) v[0][0][0][0][0][i]=true,f[0][0][0][0][0][i]=1;
printf("%lld\n",dfs(s[5],s[4],s[3],s[2],s[1],0));
return 0;
}