令f[i][j][k]表示到第i位,状态为j,第i位为k的方案数。其中状态0表示之前没有k个B和k个W;1表示只有k个B;2表示之前有k个B和k个W。转移的时候只需要得到这一位填某一个字母的情况下会不会产生新的k个连续串,然后该加的地方加上,改去掉重复方案的去掉即可。
AC代码如下:
#include<iostream>
#include<cstdio>
#include<cstring>
#define mod 1000000007
#define N 1000005
using namespace std;
int n,k,dp[N][3][2],sw[N],sb[N]; char ch[N];
void up(int &t,int x,int y,int z){
t=x+y; if (t>=mod) t-=mod;
t+=z;
if (t<0) t+=mod; else if (t>=mod) t-=mod;
}
int main(){
scanf("%d%d",&n,&k); int i,tmp;
ch[1]=getchar(); while (ch[1]<'A' || ch[1]>'Z') ch[1]=getchar();
for (i=1; i<=n; i++){
if (i>1) ch[i]=getchar();
sb[i]=sb[i-1]; sw[i]=sw[i-1];
if (ch[i]=='B') sb[i]++;
else if (ch[i]=='W') sw[i]++;
}
dp[0][0][1]=1;
for (i=1; i<=n; i++){
if (ch[i]!='W'){
if (i>=k && ch[i-k]!='B' && sw[i]==sw[i-k])
tmp=dp[i-k][0][1];
else tmp=0;
up(dp[i][0][0],dp[i-1][0][0],dp[i-1][0][1],-tmp);
up(dp[i][1][0],dp[i-1][1][0],dp[i-1][1][1],tmp);
up(dp[i][2][0],dp[i-1][2][0],dp[i-1][2][1],0);
}
if (ch[i]!='B'){
if (i>=k && ch[i-k]!='W' && sb[i]==sb[i-k])
tmp=dp[i-k][1][0];
else tmp=0;
up(dp[i][0][1],dp[i-1][0][0],dp[i-1][0][1],0);
up(dp[i][1][1],dp[i-1][1][0],dp[i-1][1][1],-tmp);
up(dp[i][2][1],dp[i-1][2][0],dp[i-1][2][1],tmp);
}
}
printf("%d\n",(dp[n][2][0]+dp[n][2][1])%mod);
return 0;
}
by lych
2016.4.6