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
解题思路:
神奇的DP技巧,首先考虑一下直接状压是肯定不能通过的,复杂度是5^15,于是转换一下思路,我们记录使用1,2,3,4,5次的颜色有多少种,因为实际上不同种类的颜色之间是等价的,所以只需要记录出现的次数,因为考虑两种油漆如果当前都能涂x次,那么他们本质是一样的。所以我们用dp[a][b][c][d][e][last]记录还剩下1,2,3,4,5次使用次数的颜色的数量为a,b,c,d,e,上一次我们那的是使用次数剩余为last的颜色,那么这次在剩余last-1次的颜色中,我们只能少拿一次,因为不能拿相同的,这样转移就很容易的表示出来了。那么很容易推出转移式子,注意对于相邻位置不能是相同的颜色的设定,我们规定这次拿的等价颜色不能喝上次的等价颜色相同,于是直接在系数里面判断一下减去即可。。这个DP我完全不会,看了这位同学的博客理解了,多谢。OTZ
代码如下:
#include <bits/stdc++.h>
using namespace std;
const int mod = 1e9+7;
int n, cnt[6];
typedef long long LL;
LL dp[16][16][16][16][16][6];
LL dfs(int a, int b, int c, int d, int e, int last){
if(a + b + c + d + e == 0) return dp[a][b][c][d][e][last] = 1;
if(~dp[a][b][c][d][e][last]) return dp[a][b][c][d][e][last];
LL res = 0;
if(a > 0) res += (a - (last == 2)) * dfs(a - 1, b, c, d, e, 1), res %= mod;
if(b > 0) res += (b - (last == 3)) * dfs(a + 1, b - 1, c, d, e, 2), res %= mod;
if(c > 0) res += (c - (last == 4)) * dfs(a, b + 1, c - 1, d, e, 3), res %= mod;
if(d > 0) res += (d - (last == 5)) * dfs(a, b, c + 1, d - 1, e, 4), res %= mod;
if(e > 0) res += e * dfs(a, b, c, d + 1, e - 1, 5), res %= mod;
return dp[a][b][c][d][e][last] = res;
}
int main(){
scanf("%d", &n);
for(int i = 1; i <= n; i++){
int x;
scanf("%d", &x);
cnt[x]++;
}
memset(dp, -1, sizeof(dp));
LL ans = dfs(cnt[1], cnt[2], cnt[3], cnt[4], cnt[5], 0);
cout << ans << endl;
return 0;
}