1079: [SCOI2008]着色方案（记忆化搜索）

本文主要是介绍1079: [SCOI2008]着色方案（记忆化搜索），希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

Description
　　有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
HINT
100%的数据满足：1 <= k <= 15, 1 <= ci <= 5

Source

话说BZOJ的很多题都是看上去清晰明了，写起来玄妙无比~~
思路：
这个题的记忆化搜索状态表示是按照这种颜色的能涂次数来算的，如果不算相邻不能相同的情况还是很好理解的。

相邻相同情况：如果上一次的的次数为2，当前涂的是次数为1的，那么要减去上一次传下来的2。因为每种颜色只要只可能存在于一个维度，所以就保证了不会有相邻重复颜色的情况。

#include <cstdio>
#include <cstring>
#include <algorithm>using namespace std;typedef long long ll;const ll MOD = 1e9 + 7;
ll c[10],vis[16][16][16][16][16][6],dp[16][16][16][16][16][6];ll dfs(ll a,ll b,ll c,ll d,ll e,ll n)
{if(vis[a][b][c][d][e][n])return dp[a][b][c][d][e][n];if(a + b + c + d + e == 0) return dp[a][b][c][d][e][n] = 1;ll ans = 0;if(a)ans += (a - (n == 2)) * dfs(a - 1,b,c,d,e,1);if(b)ans += (b - (n == 3)) * dfs(a + 1,b - 1,c,d,e,2);if(c)ans += (c - (n == 4)) * dfs(a,b + 1,c - 1,d,e,3);if(d)ans += (d - (n == 5)) * dfs(a,b,c + 1,d - 1,e,4);if(e)ans += e * dfs(a,b,c,d + 1,e - 1,5);vis[a][b][c][d][e][n] = 1;return dp[a][b][c][d][e][n] = ans % MOD;
}int main()
{int n;scanf("%d",&n);for(int i = 1;i <= n;i++){int t;scanf("%d",&t);c[t]++;}printf("%lld\n",dfs(c[1],c[2],c[3],c[4],c[5],0) % MOD);return 0;
}