bzoj1079[SCOI2008]着色方案

来源：互联网发布：ios艺术字体软件编辑：程序博客网时间：2024/05/24 06:32

不给n的范围是最sao的。。。
一开始想的是直接暴力搜，反正k才15，结果发现是O（n*2^k）妥妥的t。
然后学习了新的搜索姿势。。
就是搜索涂几次的油漆有几个，这样c<=5，很明显非常优。
所以用F[a][b][c][d][e][l]表示当前有能涂1次的油漆a个，能涂2次的b个….前一个颜色为l，再搞下转移就行了。

/**************************************************************    Problem: 1079    User: NamlessYang    Language: C++    Result: Accepted    Time:64 ms    Memory:56120 kb****************************************************************/#include<cstdio>#include<algorithm>#include<cstring>#define fo(i,a,b) for(int i=a;i<=b;i++)#define fd(i,a,b) for(int i=a;i>=b;i--)using namespace std;typedef long long ll;int n,m;const int N=1e5+5;const int mo=1e9+7;int a[10];ll f[16][16][16][16][16][6];bool bz[16][16][16][16][16][6];inline ll dfs(int a,int b,int c,int d,int e,int last){    ll t=0;    if (bz[a][b][c][d][e][last])return f[a][b][c][d][e][last];    if (a+b+c+d+e==0)return 1;    if (a)t+=(a-(last==2))*dfs(a-1,b,c,d,e,1);    if (b)t+=(b-(last==3))*dfs(a+1,b-1,c,d,e,2);    if (c)t+=(c-(last==4))*dfs(a,b+1,c-1,d,e,3);    if (d)t+=(d-(last==5))*dfs(a,b,c+1,d-1,e,4);    if (e)t+=e*dfs(a,b,c,d+1,e-1,5);    bz[a][b][c][d][e][last]=1;    return f[a][b][c][d][e][last]=(t%mo); }int main(){    scanf("%d",&n);    fo(i,1,n)    {        int x;        scanf("%d",&x);        a[x]++;    }    printf("%lld\n",dfs(a[1],a[2],a[3],a[4],a[5],0));    return 0;}

0 0