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

10

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!!!

 

 

 

posted on 2018-08-07 09:29  Frank-King  阅读(101)  评论(0编辑  收藏  举报