TC SRM551 DIV2 950 countArrangements dp

题目大意：给出由ABC三种颜色的巧克力球，发给len个人，人是围成圈的，相邻的人球的颜色不能相同，求不同分法种数mod一个数。

代码：参考了下别人写的dp。

#include <vector>
#include <cstring>
#include <list>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <stack>
#include <bitset>
#include <algorithm>
#include <functional>
#include <numeric>
#include <utility>
#include <sstream>
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <ctime>

using namespace std;

int dp[51][51][51][4];
const int mod=1000000007;

class ColorfulCupcakesDivTwo {
public:
	int countArrangements(string cupcakes)
	{
        int anum=0,bnum=0,ans=0,len=cupcakes.length();
        for(int i=0;i<len;i++)
        {
            if(cupcakes[i]=='A') anum++;
            else if(cupcakes[i]=='B') bnum++;
        }
        for(int st=1;st<=3;st++)
        {
            memset(dp,0,sizeof(dp));
            if(st==1) dp[1][1][0][1]=1;
            else if(st==2) dp[1][0][1][2]=1;
            else dp[1][0][0][3]=1;
            for(int x=2;x<=len;x++)
                for(int i=0;i<=anum;i++)
                    for(int j=0;j<=bnum;j++)
                    {
                        if(i)
                        {
                            dp[x][i][j][1]=(dp[x][i][j][1]+dp[x-1][i-1][j][3])%mod;
                            dp[x][i][j][1]=(dp[x][i][j][1]+dp[x-1][i-1][j][2])%mod;
                        }
                        if(j)
                        {
                            dp[x][i][j][2]=(dp[x][i][j][2]+dp[x-1][i][j-1][1])%mod;
                            dp[x][i][j][2]=(dp[x][i][j][2]+dp[x-1][i][j-1][3])%mod;
                        }
                        if(x-i-j)
                        {
                            dp[x][i][j][3]=(dp[x][i][j][3]+dp[x-1][i][j][1])%mod;
                            dp[x][i][j][3]=(dp[x][i][j][3]+dp[x-1][i][j][2])%mod;
                        }
                    }
            ans=(ans+(dp[len][anum][bnum][1]+dp[len][anum][bnum][2]+dp[len][anum][bnum][3]-dp[len][anum][bnum][st])%mod)%mod;
        }
        return ans%mod;
	}
};