Ural 2018. The Debut Album

给你一个长度n，和两个小于n的整数a，b

连续的1的个数不超过a个，连续的0的个数不超过b个的长为n的01串的总数，答案对10^9+9取模

-----------------------------------我是分割线---------------------------------------------------------

定义dp[bit][i][j]是以bit为结束，长为i，bit在最后的连续长度为j的串的个数，则有

#include<cstdio>
#include<cstring>
using namespace std;
#define LL long long
LL dp[2][305][2];
const int MOD=1e9+7;

int main(){
    int n,a,b;
    while(~scanf("%d %d %d",&n,&a,&b)){
        memset(dp,0,sizeof(dp));
        dp[0][1][1]=dp[1][1][1]=1;
        for(int i=2;i<=n;i++){
            dp[1][1][i%2]=dp[0][1][i%2]=0;
            for(int j=2;j<=a;j++){
                dp[0][j][i%2]=dp[0][j-1][(i-1)%2];
                dp[1][1][i%2]=(dp[1][1][i%2]%MOD+dp[0][j-1][(i-1)%2]%MOD)%MOD;
            }
            dp[1][1][i%2]=(dp[1][1][i%2]%MOD+dp[0][a][(i-1)%2]%MOD)%MOD;
            for(int j=2;j<=b;j++){
                dp[1][j][i%2]=dp[1][j-1][(i-1)%2];
                dp[0][1][i%2]=(dp[0][1][i%2]%MOD+dp[1][j-1][(i-1)%2]%MOD)%MOD;
            }
            dp[0][1][i%2]=(dp[0][1][i%2]%MOD+dp[1][b][(i-1)%2]%MOD)%MOD;
        }

//        for(int i=1;i<=a;i++)
//            for(int j=1;j<=n;j++)
//                printf(j<n?"%d ":"%d\n",dp[0][i][j]);
//        puts("");
//        for(int i=1;i<=b;i++)
//            for(int j=1;j<=n;j++)
//                printf(j<n?"%d ":"%d\n",dp[1][i][j]);
        LL ans=0;
        for(int i=1;i<=a;i++)
            ans=(ans%MOD+dp[0][i][n%2]%MOD)%MOD;
        for(int i=1;i<=b;i++)
            ans=(ans%MOD+dp[1][i][n%2]%MOD)%MOD;
        printf("%lld\n",ans);
    }
    return 0;
}

以上（