题意:
把n个蛋糕(最多3种颜色)按1…n放成一圈。
要求相邻的蛋糕颜色不同,求有多少种放法。
思路:
先看基本的方程
f[i][j][k][first][last]
表示A, B, C各剩i, j, k且第一个是first, 最后一个是last的方法数。
边界:
i+j+k = 0, 判断首尾是否相等
但是,既然我们限定了第一个数,那么就可以把它从状态里面拿出来。
f[i][j][k][last]
并且在开始dp的时候根据第一个数是0, 1 or 2来初始化数组
不过这样只是节省了内存,时间还是一样的
class ColorfulCupcakesDivTwo
{
public:
int dp(int A, int B, int C, int first, int last) {
if (f[A][B][C][first][last] != -1) return f[A][B][C][first][last];
int t = A+B+C, ret = 0;
if (!t) {
if (first == last) return 0;
return 1;
}
if (A && last != 0) ret = (ret+dp(A-1, B, C, first, 0))%Mod;
if (B && last != 1) ret = (ret+dp(A, B-1, C, first, 1))%Mod;
if (C && last != 2) ret = (ret+dp(A, B, C-1, first, 2))%Mod;
f[A][B][C][first][last] = ret;
return ret;
}
int countArrangements(string cupcakes)
{
n = cupcakes.length();
a = b = c = 0;
rep(i, 0, n-1) {
if (cupcakes[i] == 'A') ++a;
else if (cupcakes[i] == 'B') ++b;
else ++c;
}
memset(f, -1, sizeof(f));
int ans = 0;
if (a) ans = (ans+dp(a-1, b, c, 0, 0))%Mod;
if (b) ans = (ans+dp(a, b-1, c, 1, 1))%Mod;
if (c) ans = (ans+dp(a, b, c-1, 2, 2))%Mod;
return ans;
}
};