BZOJ 1079 [SCOI2008]着色方案 - 花式DP

最新推荐文章于 2019-02-11 16:48:47 发布

x_1023

最新推荐文章于 2019-02-11 16:48:47 发布

阅读量276

点赞数

分类专栏：一般DP

本文链接：https://blog.csdn.net/x_1023/article/details/78380990

版权

一般DP 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

从未见过DP还可以这样玩的，一开始以为是数学，C+容斥，后来觉得不对，哪有数学题长成这个样的，然后各种dp，15维的dp实在是没敢写，于是膜了hzwer的代码。。。真长见识。

考虑到c[i]的范围不超过5，其中必有玄机，于是可以用5维表示还可以涂i块木块的颜色种数。因为对于两种均可以涂i块的颜色，除了考虑与上一状态不能重复以外，是完全等价的。排除上一状态影响，则将当前状态的转移的贡献减去一份即可。考虑到dp方程的转移方向太过于复杂，用记忆化搜索会好一些。

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<algorithm>

using namespace std;

const int maxn=18;
const int mod=1000000007;

int n;
int c[maxn],num[maxn];
int dp[maxn][maxn][maxn][maxn][maxn][maxn];

int memorial_search(int a,int b,int c,int d,int e,int last)
{
    int &res=dp[a][b][c][d][e][last];
    if(res!=-1)return res;
    if(a+b+c+d+e==0)return res=1;
    res=0;
    if(a)
        res=(res+1LL*(a-(last==2))*memorial_search(a-1,b,c,d,e,1))%mod;
    if(b)
        res=(res+1LL*(b-(last==3))*memorial_search(a+1,b-1,c,d,e,2))%mod;
    if(c)
        res=(res+1LL*(c-(last==4))*memorial_search(a,b+1,c-1,d,e,3))%mod;
    if(d)
        res=(res+1LL*(d-(last==5))*memorial_search(a,b,c+1,d-1,e,4))%mod;
    if(e)
        res=(res+1LL*e*memorial_search(a,b,c,d+1,e-1,5))%mod;
    return res;
}
int main()
{
    memset(dp,-1,sizeof dp);
    scanf("%d",&n);
    for(int i=1,now;i<=n;i++)
        scanf("%d",&now),num[now]++;
    printf("%d",memorial_search(num[1],num[2],num[3],num[4],num[5],-1));
    return 0;
}