题解
这题是DP,一开始的想法是写一个5^15的dp,但是肯定会超时,所以就要换一个定义。
我们开dp数组 f[a][b][c][d][e][last] 记录剩下1,2,3,4,5次可涂色次数的颜色的油漆种数为a,b,c,d,e,上一次我们涂的是使用次数剩余为last的颜料。这次在剩余last-1次的颜料中,我们只能少拿一次,那么就可以推出转移式子,用记忆化dfs求解。
#include<cstdio>
#include<string>
#include<cstring>
using namespace std;
const int maxn=16,tt=1e9+7;
int n,m,a[maxn];
long long f[maxn][maxn][maxn][maxn][maxn][6];
long long dfs(int a,int b,int c,int d,int e,int last){
if (a+b+c+d+e==0) return f[a][b][c][d][e][last]=1ll;
if (f[a][b][c][d][e][last]!=-1ll) return f[a][b][c][d][e][last];
long long sum=0;
if (a>0) sum+=(a-(last==2))*dfs(a-1,b,c,d,e,1),sum%=tt;
if (b>0) sum+=(b-(last==3))*dfs(a+1,b-1,c,d,e,2),sum%=tt;
if (c>0) sum+=(c-(last==4))*dfs(a,b+1,c-1,d,e,3),sum%=tt;
if (d>0) sum+=(d-(last==5))*dfs(a,b,c+1,d-1,e,4),sum%=tt;
if (e>0) sum+=e*dfs(a,b,c,d+1,e-1,5),sum%=tt;
f[a][b][c][d][e][last]=sum; return sum;
}
int main(){
scanf("%d",&m); int x;
for (int i=1;i<=m;i++) scanf("%d",&x),a[x]++;
memset(f,-1,sizeof(f));
printf("%lld\n",dfs(a[1],a[2],a[3],a[4],a[5],0));
return 0;
}