给你一个长度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;
}
以上(