dp[][][][] 存方法数
第一维 存第一个放黑球还是白球
第二维 存第i个球
第三维 存第i个球是黑还是白
第四维 存第i个球前连续几个球是相同颜色
要减去 第n-i 个球的时候开头有i个相同颜色的求
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;
#include <queue>
#include <float.h>
#include <stack>
#include <vector>
#include <bitset>
#include <deque>
#include <set>
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <map>
typedef long long LL;
typedef unsigned long long ULL;
const int INF = 1<<29;
const LL mod = 1e9+7;
const int MAXN = 110100;//点数的最大值
const int MAXM = 520;//边数的最大值
const double pi = acos(-1.0);
inline void Mod(int &x)
{
while(x>=2015)
x-=2015;
}
int dp[2][MAXN][2][7];
int main()
{
memset(dp,0,sizeof(dp));
dp[0][1][0][0]=dp[1][1][1][0]=1;
for(int i=2; i<=100000; i++)
{
for(int j=0; j<=5&&j+1<=i; j++)
{
if(j!=0)//加相同颜色
{
dp[1][i][0][j]+=dp[1][i-1][0][j-1];
dp[0][i][0][j]+=dp[0][i-1][0][j-1];
Mod(dp[0][i][0][j]);
Mod(dp[1][i][0][j]);
dp[1][i][1][j]+=dp[1][i-1][1][j-1];
dp[0][i][1][j]+=dp[0][i-1][1][j-1];
Mod(dp[0][i][1][j]);
Mod(dp[1][i][1][j]);
}
//不同颜色
dp[0][i][0][0]+=dp[0][i-1][1][j];
Mod(dp[0][i][0][0]);
dp[1][i][0][0]+=dp[1][i-1][1][j];
Mod(dp[1][i][0][0]);
dp[0][i][1][0]+=dp[0][i-1][0][j];
Mod(dp[0][i][1][0]);
dp[1][i][1][0]+=dp[1][i-1][0][j];
Mod(dp[1][i][1][0]);
}
}
int n,t,cas=0;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
int ans=0;
for(int i=0; i<=5; i++)
{
ans+=dp[0][n][0][i];
Mod(ans);
ans+=dp[1][n][0][i];
Mod(ans);
ans+=dp[0][n][1][i];
Mod(ans);
ans+=dp[1][n][1][i];
Mod(ans);
}
for(int i=1; i<=6&&i<=n; i++)
{
for(int j=1; j<=i; j++)
{
if(6-j+i+1<=6)
continue;
// cout<<n-i<<" "<<6-j<<endl;
ans-=dp[1][n-i][0][6-j];
ans=(ans+2015)%2015;
ans-=dp[0][n-i][1][6-j];
ans=(ans+2015)%2015;
// cout<<ans<<endl;
}
}
printf("Case #%d: %d\n",++cas,ans);
}
return 0;
}
/*
*/