当前位置: 代码迷 >> 综合 >> BZOJ 1079 [SCOI2008] 着色方案
  详细解决方案

BZOJ 1079 [SCOI2008] 着色方案

热度:8   发布时间:2024-01-19 02:11:14.0

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

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

记忆化搜索~

剩余数目相同的颜色之间没有本质区别,所以我们只要记录每种剩余颜色数目对应的颜色种类数就可以了。

用f[x][y][z][v][k][w]表示剩余1,2,3,4,5个的颜色分别有x,y,z,v,k,w种,前一个位置填还有w数目的颜色的种类数,那么直接记忆化搜索就可以了。

注意:

1.判断的时候要判断是否还有xyzvk是否为0;

2.x-(w==2)而不是x=(w==1),因为前一个颜色填入的时候还没有减去1;

3.开long long!


#include<cstdio>
#include<iostream>
using namespace std;
#define ll long long
#define modd 1000000007int n,a[6];
ll f[16][16][16][16][16][6];
bool b[16][16][16][16][16][6];int read()
{int x=0,f=1;char ch=getchar();while(ch<'0' || ch>'9') {if(ch=='-') f=-1;ch=getchar();}while(ch>='0' && ch<='9') {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}return x*f;
}ll dfs(int x,int y,int z,int v,int k,int w)
{if(b[x][y][z][v][k][w]) return f[x][y][z][v][k][w];if(!(x+y+z+v+k)) return 1;ll tot=0;if(x) (tot+=(x-(w==2))*dfs(x-1,y,z,v,k,1))%=modd;if(y) (tot+=(y-(w==3))*dfs(x+1,y-1,z,v,k,2))%=modd;if(z) (tot+=(z-(w==4))*dfs(x,y+1,z-1,v,k,3))%=modd;if(v) (tot+=(v-(w==5))*dfs(x,y,z+1,v-1,k,4))%=modd;if(k) (tot+=k*dfs(x,y,z,v+1,k-1,5))%=modd;b[x][y][z][v][k][w]=1;return f[x][y][z][v][k][w]=tot;
}int main()
{n=read();for(int i=1;i<=n;i++) a[read()]++;printf("%lld\n",dfs(a[1],a[2],a[3],a[4],a[5],0));return 0;
}