ZOJ 3725 Painting Storages（DP解排列计数）

题目链接：Click here~~

题意：

n个格子排成一条直线，可以选择涂成红色或蓝色，问最少 m 个连续为红色的方案数。

解题思路：

应该是这次 ZOJ 月赛最水的一题，可惜还是没想到。。。

dp[i] 表示前 i 个最少 m 个连续为红色的方案数。

转移时，分类讨论：

1、前 i-1 个已经满足这个性质，那么，第 i 个随意涂色，方案数为 dp[i-1] * 2 。

2、前 i-1 个不满足这个性质，那么，要想成为方案，区间 [i-m+1,i] 必须涂成红色。并且，下标为 i-m 这个点必须是蓝色，否则就与 情况1 重复了。

而且正是由于这一点，只要剩下的区间 [1,i-m-1] 不满足这个性质，就能保证整个区间 [1,i-1] 不满足这个性质。方案数为 2^(i-m-1) - dp[i-m-1]。

#include <stdio.h>
#include <string.h>
#include <algorithm>

using namespace std;

const int N = 1e5 + 5;
const int mod = 1e9 + 7;

int dp[N],pow[N]={1};

int main()
{
    int n,m;
    for(int i=1;i<N;i++)
        pow[i] = pow[i-1] * 2 % mod;
    while(~scanf("%d%d",&n,&m))
    {
        memset(dp,0,sizeof(int)*m);
        dp[m] = 1;
        for(int i=m+1;i<=n;i++)
            dp[i] = ((dp[i-1] * 2 % mod + pow[i-m-1] - dp[i-m-1]) % mod + mod) % mod;
        printf("%d\n",dp[n]);
    }
    return 0;
}