bzoj1079--记忆化搜索

时间:2024-01-17 11:52:08

题目大意:
有n个木块排成一行,从左到右依次编号为1~n。你有k种颜色的油漆,其中第i种颜色的油漆足够涂ci个木块。
所有油漆刚好足够涂满所有木块,即c1+c2+...+ck=n。相邻两个木块涂相同色显得很难看,所以你希望统计任意两
个相邻木块颜色不同的着色方案。

题解:
看到数据范围第一个想到的就是dp。但5^15显然不现实。注意到ci相等的颜色本质上是相同的,于是可以记忆化搜索。

时间复杂度:O(15^5)

代码:

 #include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define mod 1000000007
#define ll long long
int i,j,k,n,m,s[];
ll f[][][][][][];
inline ll Dfs(int a,int b,int c,int d,int e,int l){
ll Sum=;
if(f[a][b][c][d][e][l])return f[a][b][c][d][e][l];
if(a+b+c+d+e==)return ;
if(a)Sum+=(a-(l==))*Dfs(a-,b,c,d,e,);
if(b)Sum+=(b-(l==))*Dfs(a+,b-,c,d,e,);
if(c)Sum+=(c-(l==))*Dfs(a,b+,c-,d,e,);
if(d)Sum+=(d-(l==))*Dfs(a,b,c+,d-,e,);
if(e)Sum+=e*Dfs(a,b,c,d+,e-,);
return f[a][b][c][d][e][l]=Sum%mod;
}
int main()
{
scanf("%d",&n);
for(i=;i<=n;i++){
scanf("%d",&k);
s[k]++;
}
printf("%lld",Dfs(s[],s[],s[],s[],s[],));
return ;
}

bzoj1079