foj2204 Problem 2204 7 dp

Problem 2204 7
Accept: 50 Submit: 142
Time Limit: 2000 mSec Memory Limit : 65536 KB
Problem Description
n个有标号的球围成一个圈。每个球有两种颜色可以选择黑或白染色。问有多少种方案使得没有出现连续白球7个或连续黑球7个。

Input
第一行有多组数据。第一行T表示组数。(T <= 20)

每组包含n，表示球的个数。(1 <= n <= 100000)

Output
每组先输出 “Case #x: ” (其中x为当前组数) 该行接下来输出方案数。方案数mod 2015。

Sample Input
2
7
1
Sample Output
Case #1: 126
Case #2: 2
Source
FOJ有奖月赛-2015年10月

Submit Back Status Discuss
先化环为线，先只考虑线的情况。
dp,用dp[i][j][jf][k][kf],jf kf 为0 表示白，为1表示黑。整个式子表示，有i个球，前6个球中是jf有j个，kf的有k个，则，当不足7个时，向后加入一个球，可以加入黑，也可以加入白球，有两种方法选择。状态转移就是，可以加入一个与最后球不同色的球，或者，加入一个相同色的球，只要不连续7个就可以了。
再来考虑环的问题，只要前后如果同色，前6个与后6个，加起来同色的不超过7就可以了。
总的时间复杂度为o(n * 7 * 7 * 2 * 2),大概也可以ac了。

#define N 205
#define M 100005
#define maxn 205
#define MOD 1000000000000000007
int n,T;
short dp[100005][7][2][7][2];
int p[20],all;
int main()
{
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
    fill(dp,0);
    dp[1][1][0][1][0] = 1;
    dp[1][1][1][1][1] = 1;
    For(i,1,100000){
        For(j,1,7)
            For(jf,0,2){
                for(int k = 1;k < 7;k++){
                    For(kf,0,2){
                        if(i <= 5){
                            if(j == i && k == i && jf == kf){
                                dp[i+1][j+1][jf][k+1][kf] = (dp[i+1][j+1][jf][k+1][kf] + dp[i][j][jf][k][kf]) % mod;
                                dp[i+1][j][jf][1][kf ^ 1] = (dp[i+1][j][jf][1][kf ^ 1] + dp[i][j][jf][k][kf]) % mod;
                            }
                            else {
                                dp[i+1][j][jf][k+1][kf] = (dp[i+1][j][jf][k+1][kf] + dp[i][j][jf][k][kf]) % mod;
                                dp[i+1][j][jf][1][kf ^ 1] = (dp[i+1][j][jf][1][kf ^ 1] + dp[i][j][jf][k][kf]) % mod;
                            }
                        }
                        else {
                            if(j < 7 && k + 1 < 7)
                                dp[i+1][j][jf][k+1][kf] = (dp[i+1][j][jf][k+1][kf] + dp[i][j][jf][k][kf]) % mod;
                            if(j < 7)
                                dp[i+1][j][jf][1][kf ^ 1] = (dp[i+1][j][jf][1][kf ^ 1] + dp[i][j][jf][k][kf]) % mod;
                        }
                    }
                }
            }
    }
     while(S(T)!=EOF)
    {
        int i = 0,ans;
        while(T--){
            S(n);
            ans = 0;
            For(j,1,7){
                For(jf,0,2){
                    For(k,1,7){
                        For(kf,0,2){
                            if(jf == kf){
                                if(min(j + k,n)< 7)
                                    ans = (ans + dp[n][j][jf][k][kf]) % mod;
                            }
                            else
                                ans = (ans + dp[n][j][jf][k][kf]) % mod;
                        }
                    }
                }
            }
            printf("Case #%d: %d\n",++i,ans);
        }
    }
    //fclose(stdin);
    //fclose(stdout);
    return 0;
}