bzoj 1079(记忆化搜索)

传送门
题意:略

这个状态定义简直让人orzorzorz。
dp[a][b][c][d][e][last]表示能涂1个的还有a种,能涂2个还有b种……能涂5个的还有e种,上一次选的是能涂last个的,进行记忆化搜索即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll MOD=1e9+7;
ll dp[16][16][16][16][16][16];
int num[6],n;
inline int read() {
    int x=0;char c=getchar();
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x;
}
ll f(int a,int b,int c,int d,int e,int last) {
    if ((a|b|c|d|e)==0) return 1;
    if (dp[a][b][c][d][e][last]) return dp[a][b][c][d][e][last];
    ll ret=0;
    if (a) ret+=f(a-1,b,c,d,e,1)*(a-(last==2)),ret%=MOD;
    if (b) ret+=f(a+1,b-1,c,d,e,2)*(b-(last==3)),ret%=MOD;
    if (c) ret+=f(a,b+1,c-1,d,e,3)*(c-(last==4)),ret%=MOD;
    if (d) ret+=f(a,b,c+1,d-1,e,4)*(d-(last==5)),ret%=MOD;
    if (e) ret+=f(a,b,c,d+1,e-1,5)*e,ret%=MOD;
    return dp[a][b][c][d][e][last]=ret;
}
int main() {
//  freopen("bzoj 1079","r",stdin);
    memset(dp,0,sizeof(dp));
    n=read();
    for (int i=1;i<=n;++i) ++num[read()];
    printf("%lld\n",f(num[1],num[2],num[3],num[4],num[5],0));
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值