SRM 551 div2 950(DP, 环形+优化)

题意：
把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;
        }
};