LUOGU P2476 [SCOI2008]着色方案

传送门

解题思路

毒瘤题，，刚开始写了个奇奇怪怪的哈希，结果T了5个点。。后来深(kan)入(le)思(ti)考(jie)，发现c的范围很小，设$f[a][b][c][d][e][pre]​$表示还能涂一个格子的有a个，两个格子的有b个。。。pre表示上一个涂的颜色，转移看代码，比较好想。

#include<iostream>
#include<cstdio>
#include<cstring>

using namespace std;
const int MAXN = ;
const int mod = 1e9+;
typedef long long LL;

int k,c[MAXN],cnt[MAXN];
int f[MAXN][MAXN][MAXN][MAXN][MAXN][];

int dfs(int a,int b,int c,int d,int e,int pre){
    if(f[a][b][c][d][e][pre]) return f[a][b][c][d][e][pre];
    if(!a && !b && !c && !d && !e) return f[a][b][c][d][e][pre]=;
    int ret=;
    if(a) ret=(ret+(LL)(a-(pre==))*dfs(a-,b,c,d,e,)%mod)%mod;
    if(b) ret=(ret+(LL)(b-(pre==))*dfs(a+,b-,c,d,e,)%mod)%mod;
    if(c) ret=(ret+(LL)(c-(pre==))*dfs(a,b+,c-,d,e,)%mod)%mod;
    if(d) ret=(ret+(LL)(d-(pre==))*dfs(a,b,c+,d-,e,)%mod)%mod;
    if(e) ret=(ret+(LL)e*dfs(a,b,c,d+,e-,)%mod)%mod;
    return f[a][b][c][d][e][pre]=ret;
}

int main(){
    scanf("%d",&k);
    for(int i=;i<=k;i++) scanf("%d",&c[i]),cnt[c[i]]++;
    printf("%d",dfs(cnt[],cnt[],cnt[],cnt[],cnt[],));
    return ;
}