BZOJ1079或洛谷2476 [SCOI2008]着色方案

一道记忆化搜索

BZOJ原题链接

洛谷原题链接

发现对于能涂木块数量一样的颜色在本质上是一样的，所以可以直接压在一个状态，而这题的数据很小，直接暴力开\(6\)维。
定义\(f[a][b][c][d][e][la]\)，\(a\)表示能涂\(1\)个木块的颜色总数，\(b\)表示能涂\(2\)个木块的颜色总数，\(c,d,e\)同理，\(la\)表示上次涂的颜色是能涂\(la\)个木块的。
然后考虑状态转移。如果用能涂\(1\)个木块的颜色去涂，则状态由\((a-(la==2))*f[a-1][b][c][d][e][1]\)转移来，因为有\(a\)种颜色，每一种都可以涂，所以要乘上\(a\)，但注意当前状态的\(la=2\)的情况，就是说这个状态是由可以涂\(2\)个木块的颜色转移过来的，这时原来能涂\(2\)个木块的颜色变成只能涂\(1\)个了，题目要求相邻木块不能涂同一颜色，所以要将\(a\)减去\(1\)。对于用能涂\(2\)个木块的颜色去涂，则状态由\((b-(la==3))*f[a+1][b-1][c][d][e][2]\)转移来，其他同理得。
显然使用记忆化搜索更好打。

#include<cstdio>
using namespace std;
typedef long long ll;
const int N = 16;
const int mod = 1e9 + 7;
ll f[N][N][N][N][N][6];
int co[6];
int re()
{
	int x = 0;
	char c = getchar();
	bool p = 0;
	for (; c<'0' || c>'9'; c = getchar())
		p = (c == '-' || p) ? 1 : 0;
	for (; c >= '0'&&c <= '9'; c = getchar())
		x = x * 10 + (c - '0');
	return p ? -x : x;
}
ll dp(int a, int b, int c, int d, int e, int la)
{
	ll s = 0, &k = f[a][b][c][d][e][la];
	if (k)
		return k;
	if (!(a | b | c | d | e))
		return 1;
	if (a)
		s += 1LL * (a - (la == 2))*dp(a - 1, b, c, d, e, 1);
	if (b)
		s += 1LL * (b - (la == 3))*dp(a + 1, b - 1, c, d, e, 2);
	if (c)
		s += 1LL * (c - (la == 4))*dp(a, b + 1, c - 1, d, e, 3);
	if (d)
		s += 1LL * (d - (la == 5))*dp(a, b, c + 1, d - 1, e, 4);
	if (e)
		s += 1LL * e*dp(a, b, c, d + 1, e - 1, 5);
	k = s % mod;
	return k;
}
int main()
{
	int i, n;
	n = re();
	for (i = 1; i <= n; i++)
		co[re()]++;
	printf("%lld", dp(co[1], co[2], co[3], co[4], co[5], 0));
	return 0;
}