[BZOJ1079][SCOI2008]着色方案（记搜）

题目描述

传送门

题目大意：有n个木块排成一行，从左到右依次编号为1~n。你有k种颜色的油漆，其中第i种颜色的油漆足够涂ci个木块。所有油漆刚好足够涂满所有木块，即c1+c2+…+ck=n。统计任意两个相邻木块颜色不同的着色方案数，对1000000007取模。

题解

f(a,b,c,d,e,g)表示剩余1,2,3,4,5个的颜色分别有a,b,c,d,e个，上一次是从剩余g个的颜色里选择了一个 的方案数，这样的话判断相邻两个不能一样然后记搜就行了

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define LL long long
#define Mod 1000000007
#define N 16

int k,cnt[N];
long long f[N][N][N][N][N][6];

long long dp(int a,int b,int c,int d,int e,int g)
{
    if (f[a][b][c][d][e][g]!=-1) return f[a][b][c][d][e][g];
    long long *re=&f[a][b][c][d][e][g];*re=0;
    if (a) *re+=(a-(g==2?1:0))*dp(a-1,b,c,d,e,1)%Mod;
    if (b) *re+=(b-(g==3?1:0))*dp(a+1,b-1,c,d,e,2)%Mod;
    if (c) *re+=(c-(g==4?1:0))*dp(a,b+1,c-1,d,e,3)%Mod;
    if (d) *re+=(d-(g==5?1:0))*dp(a,b,c+1,d-1,e,4)%Mod;
    if (e) *re+=e*dp(a,b,c,d+1,e-1,5)%Mod;
    *re%=Mod;
    return *re;
}
int main()
{
    scanf("%d",&k);
    for (int i=1;i<=k;++i)
    {
        int x;scanf("%d",&x);
        ++cnt[x];
    }
    memset(f,-1,sizeof(f));
    for (int i=1;i<=5;++i) f[0][0][0][0][0][i]=1;
    printf("%lld\n",dp(cnt[1],cnt[2],cnt[3],cnt[4],cnt[5],0));
}