SRM 551 ColorfulCupcakesDivTwo

有n个蛋糕(1<=n<=50)，蛋糕有A,B,C三种。（读入一个字符串，由‘A’、’B‘、’C‘三种字符组成）现在有n个人围成一圈，每个人将得到一个蛋糕，而且相邻两个人的蛋糕是不同的。求有多少种分法？答案mod 1e9+7。

思路

<1>对于一个状态如何转移：我们发现可以从一个X->可以以其为中心加上其他字符来实现转移

int dfs(int f,int a,int b,int c,int t){
    if(a+b+c==A+B+C)return f!=t;
    int &ans=dp[f][a][b][c][t];
    if(~ans)return ans;ans=0;
    if(a<A&&f!=0)add(ans,dfs(0,a+1,b,c,t));
    if(b<B&&f!=1)add(ans,dfs(1,a,b+1,c,t));
    if(c<C&&f!=2)add(ans,dfs(2,a,b,c+1,t));
    return ans;
}

//枚举中心
memset(dp,-1,sizeof(dp));
if(A)add(ans,dfs(0,1,0,0,0));
memset(dp,-1,sizeof(dp));
if(B)add(ans,dfs(1,0,1,0,1));
memset(dp,-1,sizeof(dp));
if(C)add(ans,dfs(2,0,0,1,2));
return ans;