【DP】 FZU 2204 7 成环

点击打开链接

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;
}
/*
*/