2018. The Debut Album

http://acm.timus.ru/problem.aspx?space=1&num=2018

真心爱过，怎么能彻底忘掉

题目大意：

长度为n的串，由1和2组成，连续的1不能超过a个，连续的2不能超过b个

dpa[i] 表示长度为i时以a为结尾的串的个数，dpb[i] 类似

求dpa[i]时 需要枚举结尾a的个数就可以了 dpb[i] 类似

#include <iostream>

#include <stdio.h>

#include <stdlib.h>

#include <io.h>

#include <string.h>



using namespace std;



const int N=50001;

const int M=301;

const unsigned int MOD=1000000007;

unsigned int dpa[N];

unsigned int dpb[N];

int main()

{

    //freopen("data.in","r",stdin);

    memset(dpa,0,sizeof(dpa));

    memset(dpb,0,sizeof(dpb));

    int n,a,b;

    cin>>n>>a>>b;

    dpa[0]=dpb[0]=1;

    for(int i=1;i<=n;++i)

    {

        for(int j=1;j<=a&&j<=i;++j)

        {

            dpa[i]=(dpa[i]+dpb[i-j])%MOD;

        }

        for(int j=1;j<=b&&j<=i;++j)

        {

            dpb[i]=(dpb[i]+dpa[i-j])%MOD;

        }

    }

    cout<<((dpa[n]+dpb[n])%MOD)<<endl;

    return 0;

}