题目大意:给出由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;
}
};