BZOJ 1079 [SCOI2008]着色方案【记忆化搜索】

BZOJ 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

HINT

100%的数据满足：1 <= k <= 15, 1 <= ci <= 5

题解：

直接集合（状压）dp肯定是不行的，空间爆！！！考虑两种油漆如果当前都能涂x次，那么他们本质是一样的。

上记忆化搜索（%hzwer）

f[a][b][c][d][e][pre] 表示只能涂一次的油漆有 a 个，只能涂2次的油漆有 b 个， …… 前一种颜色为 pre。

然后通过乘法原理弄粗结果。。。注意要特判相邻的两个颜色！！！

代码：

 1 #include<bits/stdc++.h>
 2 #define ll long long
 3 const int mo=1e9+7;
 4 using namespace std;
 5 ll f[16][16][16][16][16][6];
 6 int x[6],n;
 7 bool vis[16][16][16][16][16][6];
 8 inline ll doing(int a,int b,int c,int d,int e,int pre)
 9 {
10     ll tot=0;
11     if (vis[a][b][c][d][e][pre]) return f[a][b][c][d][e][pre];
12     if (a+b+c+d+e==0) return 1;
13     if (a) tot+=(a-(pre==2))*doing(a-1,b,c,d,e,1);
14     if (b) tot+=(b-(pre==3))*doing(a+1,b-1,c,d,e,2);
15     if (c) tot+=(c-(pre==4))*doing(a,b+1,c-1,d,e,3);
16     if (d) tot+=(d-(pre==5))*doing(a,b,c+1,d-1,e,4);
17     if (e) tot+=e*doing(a,b,c,d+1,e-1,5);
18     vis[a][b][c][d][e][pre]=1;
19     return f[a][b][c][d][e][pre]=(tot%mo);
20 }
21 int main()
22 {
23     scanf("%d",&n);
24     for (int i=1; i<=n; i++)
25     {
26         int y;
27         scanf("%d",&y);
28         x[y]++;
29     }
30     printf("%lld",doing(x[1],x[2],x[3],x[4],x[5],0));
31     return 0;
32 }

View Code

加油加油加油！！！fighting fighting fighting!!!

BZOJ 1079 [SCOI2008]着色方案 【记忆化搜索】