1079: [SCOI2008]着色方案 记忆化搜索+DP

有n个木块排成一行，从左到右依次编号为1~n。你有k种颜色的油漆，其中第i种颜色的油漆足够涂ci个木块。
所有油漆刚好足够涂满所有木块，即c1+c2+...+ck=n。相邻两个木块涂相同色显得很难看，所以你希望统计任意两
个相邻木块颜色不同的着色方案。

Input

　　第一行为一个正整数k，第二行包含k个整数c1, c2, ... , ck。

Output

　　输出一个整数，即方案总数模1,000,000,007的结果。

Sample Input

3
1 2 3

Sample Output

10

 

一看还以为是个数学题。不会做。看了网上的题解。

emmmm

f[a][b][c][d][e] 代表剩余的元素中，能涂一个格子的颜色有a个，能涂两个格子的颜色有 b 个，能涂三个格子的颜色有 c 个，

依次类推。

递推关系就是

f[a][b][c][d][e] = dp(a-1,b,c,d,e,1)*a + dp(a+1,b-1,c,d,e,2)*b + dp(a,b+1,c-1,d,e,3)*c + dp(a,b,c+1,d-1,e,4)*d + dp(a,b,c,d+1,e-1,5)*e;

 

当前这个状态可以从前一个状态推过来。 比如我当前要用 可以染三个格子的颜色，那么就可以从 染2个格子的颜色推过来。

染三个格子的颜色，染了一个格子，就只能染两个格子了，所以  c-1 b+1.   在乘上 c，因为有c中颜色。

 

这样我们没有考虑相邻的情况，所以我们在开一维记录上一次我们用了什么颜色。

如果上次用了 可以染四个格子的颜色，那么 c 就会+1，这个时候我们不能直接 乘上 c 因为c 多 1.所以要判断上一次的颜色。

这个题用了记忆化。防止重复计算。

#include <bits/stdc++.h>
#define mem(x,v) memset(x,v,sizeof(x)) 
using namespace std;
typedef long long LL;
const long long P = 1e9+7 ;
LL f[17][17][17][17][17][6];
 
LL dp(int a, int b, int c, int d, int e, int last){
    if ((a|b|c|d|e) == 0) return 1;
    if (f[a][b][c][d][e][last]) return f[a][b][c][d][e][last];
    LL res = 0;
    if (a) res += dp(a-1,b,c,d,e,1)*(a-(last == 2));
    if (b) res += dp(a+1,b-1,c,d,e,2)*(b-(last == 3));
    if (c) res += dp(a,b+1,c-1,d,e,3)*(c-(last == 4));
    if (d) res += dp(a,b,c+1,d-1,e,4)*(d-(last == 5));
    if (e) res += dp(a,b,c,d+1,e-1,5)*e;
    return f[a][b][c][d][e][last] = res % P;
}
inline const int getint() { int r=0, k=1; char c=getchar(); for(; c<'0'||c>'9'; c=getchar()) if(c=='-') k=-1; for(; c>='0'&&c<='9'; c=getchar()) r=r*10+c-'0'; return k*r; }
 
int a[6];
int main(){
    int n;
    n = getint();
    for (int i = 0; i < n; i++)  a[getint()]++;
    LL ans = dp(a[1],a[2],a[3],a[4],a[5],0);
    printf("%lld\n",ans);
    return 0;
}