DP —— 玲珑学院OJ 1091

• 题目链接： http://www.ifrog.cc/acm/problem/1091

• 题目： 给出一个长度为N的长条（有N格），每一格可以放黑棋子或白棋子，同时不可以连续放a个黑棋或b个白棋，求一共有多少种放置的方法

• 分析：嗨呀，好气，一开始觉得可以直接算出来，用容斥+隔板法瞎推了好久，其中有一种情况非常难算：长条上放了a个黑棋子和b个白棋子的放置方法。最后题解是用动态规划来做的，因为前N个位置的合法摆放数可以被分成2类，一类是最后为黑棋，一类是最后为白棋，通过这个可以进一步推出下一个状态。

• 状态： DP[i][j] 表示 前i个位置末尾棋子为j颜色的合法摆放数。j=1表示黑棋，j=2表示白棋，用 j=0 存储2者和，所以DP[0][0] = 1

• 转移：

DP[i][1] = DP[i-1][0] - DP[i-a][0] + DP[i-a][1] 
//在i-1位置后放一个黑棋，那么i个位置的合法数量为i-1个位置的合法数量减去i-a个位置的合法数量，这时还需要加上i-a个位置末尾为黑棋的合法数量，因为这种情况导致的非法情况是不会再i-a里面统计的，否则i-1就不会合法了。
DP[i][2] = DP[i-1][0] - DP[i-b][0] + DP[i-b][2]
//同上

• AC代码：

/*************************************************************************
    > File Name: test.cpp
    > Author: Akira 
    > Mail: qaq.febr2.qaq@gmail.com 
 ************************************************************************/

#include <iostream>
#include <sstream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cstdlib>
#include <algorithm>
#include <bitset>
#include <queue>
#include <stack>
#include <map>
#include <cmath>
#include <vector>
#include <set>
#include <list>
#include <ctime>
#include <climits>
typedef long long LL;
typedef unsigned long long ULL;
typedef long double LD;
#define MST(a,b) memset(a,b,sizeof(a))
#define CLR(a) MST(a,0)
#define Sqr(a) ((a)*(a))
using namespace std;

#define MaxN 1000001
#define MaxM MaxN*10
#define INF 0x3f3f3f3f
#define bug cout<<88888888<<endl;
const int mod = 1E9+7;
template <class T>
inline void scan_d(T &ret) {
    char c; ret=0;
    while((c=getchar())<'0'||c>'9');
    while(c>='0'&&c<='9') ret=ret*10+(c-'0'),c=getchar();
}
inline void out(int x) {
   if(x>9) out(x/10);
   putchar(x%10+'0');
}
int DP[MaxN][3];
int T,a,b,n;
int sum;

inline void mod_add(int &x, int y)
{
    if((x += y) >= mod)
        x -= mod;
}
inline void mod_sub(int &x, int y)
{
    if((x -= y) < 0)
        x += mod;
}
void solve()
{
    for(int i=1;i<=n;i++)
    {
        DP[i][1] = DP[i-1][0];
        if(i>=a)
        {
            mod_sub(DP[i][1], DP[i-a][0]);
            mod_add(DP[i][1], DP[i-a][1]);
        }
        DP[i][2] = DP[i-1][0];
        if(i>=b)
        {
            mod_sub(DP[i][2], DP[i-b][0]);
            mod_add(DP[i][2], DP[i-b][2]);
        }
        DP[i][0] = DP[i][1];
        mod_add(DP[i][0],DP[i][2]);
    }
    out(DP[n][0]);
    printf("\n");
}
int main()
{
    DP[0][0] = 1;
    //std::ios::sync_with_stdio(false);
    scan_d(T);
    while(T--)
    {
        scan_d(a);
        scan_d(b);
        scan_d(n);
        sum = 0;
        solve();
    }
    //system("pause");
}