[BZOJ]1079 [SCOI2008]着色方案 记忆化搜索

1079: [SCOI2008]着色方案

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 2188   Solved: 1326
[ Submit][ Status][ Discuss]

Description

　　有n个木块排成一行，从左到右依次编号为1~n。你有k种颜色的油漆，其中第i种颜色的油漆足够涂ci个木块。
所有油漆刚好足够涂满所有木块，即c1+c2+...+ck=n。相邻两个木块涂相同色显得很难看，所以你希望统计任意两
个相邻木块颜色不同的着色方案。

Input

　　第一行为一个正整数k，第二行包含k个整数c1, c2, ... , ck。

Output

　　输出一个整数，即方案总数模1,000,000,007的结果。

Sample Input

3
1 2 3

Sample Output

10

HINT

 100%的数据满足：1 <= k <= 15, 1 <= ci <= 5

Source

[ Submit][ Status][ Discuss] 

HOME Back

这道题真的强...OI智慧无止尽啊. 今天被两道DP震惊到了， 可能是我太弱了吧. 一道是这道， 一道是NOI2011 NOI嘉年华(这道题居然用了4个DP方程...状态也好难想到). 这道题状态也是不太一样. dp[a][b][c][d][e][f]表示使用次数还剩1次的颜色(种类)个数为a， 使用次数还剩2次的颜色个数为b......使用次数还剩5次的颜色数量， 以及上一次使用的是还剩次数为f的颜色的方案数. 记忆化搜索15^6很科学， 就是很难想. 我要来写的话一定是把f作为上一次涂得什么颜色...然后就做不出来了.

考虑转移， 假如说我当前用剩余次数为1的话， 那么剩余次数为1的一共有a种， 也就是说一共可以选a种. 怎么避免跟上次相同呢 ? 如果f == 2的话， 也就是说上次用了剩余次数为2的某种颜色， 那么这次这种颜色就是剩余次数为1的了， 那我就只能.选a-1种 -- 这就是精妙所在， 我没有记录上次是什么颜色， 而是用剩余次数去判断， 反而少讨论了很多状态. 这样就可以转移了.

注意如果本次使用的是剩余次数 > 1的话， 比如说用了剩余次数为2的， 虽然b减1， 但是同时剩余次数为1的颜色也多了一种， 这样的话 a 要+ 1. 其他同理， 详见代码.

#include<stdio.h>
typedef long long dnt;
const int mod = 1e9 + 7;
int cnt[16], n, x;
dnt dp[16][16][16][16][16][16];
dnt dfs(int a, int b, int c, int d, int e, int f){
    if(!(a + b + c + d + e)) return dp[a][b][c][d][e][f] = 1;
    if(dp[a][b][c][d][e][f]) return dp[a][b][c][d][e][f];
    dnt ans = 0;
    if(a) ans += (a - (f == 2)) * dfs(a - 1, b, c, d, e, 1);
    if(b) ans += (b - (f == 3)) * dfs(a + 1, b - 1, c, d, e, 2);
    if(c) ans += (c - (f == 4)) * dfs(a, b + 1, c - 1, d, e, 3);
    if(d) ans += (d - (f == 5)) * dfs(a, b, c + 1, d - 1, e, 4);
    if(e) ans += e * dfs(a, b, c, d + 1, e - 1, 5);
    return dp[a][b][c][d][e][f] = ans % mod;
}
int main(){
    scanf("%d", &n);
    for(int i = 1; i <= n; ++i){
        scanf("%d", &x);
        cnt[x]++;
    }
    printf("%lld\n", dfs(cnt[1], cnt[2], cnt[3], cnt[4], cnt[5], 0));
}