洛谷P2476 [SCOI2008]着色方案——题解

题目传送门
题目大意：
有k种颜色的油漆，每种只能涂一定次数，所有加起来刚好能涂满，相邻的不能涂相同的颜色，求方案数。

---

思考过程：
我们发现这题的c值非常小，最大才5，于是可以用五维来暴力记录状态，再加上记录上一次涂的颜色，一共六维，可以暴力转移。

---

具体做法：
dp[t1][t2][t3][t4][t5][last]其中ti表示能涂i下的颜色有多少种，last表示上次涂的颜色，具体转移请看代码。

---

代码：

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

const long long mod=1e9+7;
long long dp[16][16][16][16][16][6];
int t[6];
int n;

long long dfs(int a,int b,int c,int d,int e,int last)
{
    if(dp[a][b][c][d][e][last]!=-1) return dp[a][b][c][d][e][last];
    if(a+b+c+d+e==0) return 1;
    long long nowans=0;
    if(a) nowans=(nowans+(a-(last==2))*dfs(a-1,b,c,d,e,1))%mod;
    if(b) nowans=(nowans+(b-(last==3))*dfs(a+1,b-1,c,d,e,2))%mod;
    if(c) nowans=(nowans+(c-(last==4))*dfs(a,b+1,c-1,d,e,3))%mod;
    if(d) nowans=(nowans+(d-(last==5))*dfs(a,b,c+1,d-1,e,4))%mod;
    if(e) nowans=(nowans+e*dfs(a,b,c,d+1,e-1,5))%mod;
    dp[a][b][c][d][e][last]=nowans; 
    return dp[a][b][c][d][e][last];
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        t[x]++; 
    }
    memset(dp,-1,sizeof(dp));
    printf("%lld\n",dfs(t[1],t[2],t[3],t[4],t[5],0));
    return 0;   
}