BZOJ1079: [SCOI2008]着色方案

题解

这题是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;
}