题目链接
https://www.lydsy.com/JudgeOnline/problem.php?id=2958
https://www.lydsy.com/JudgeOnline/problem.php?id=3269
双倍经验题,连题目名字都是一样的……
题解
设表示前个位置,当前位置填’W’/‘B’,两个条件都没有满足/第一个条件满足而第二个不满足/两个都满足 的方案数,转移就非常显然了。
代码
#include <cstdio>
#include <algorithm>
int read()
{
int x=0,f=1;
char ch=getchar();
while((ch<'0')||(ch>'9'))
{
if(ch=='-')
{
f=-f;
}
ch=getchar();
}
while((ch>='0')&&(ch<='9'))
{
x=x*10+ch-'0';
ch=getchar();
}
return x*f;
}
const int maxn=1000000;
const int mod=1000000007;
int n,k,f[maxn+10][2][3],sum[maxn+10][2][3];
char s[maxn+10];
int getsum(int a,int b,int j,int l)
{
int ans=sum[b][j][l];
if(a>0)
{
ans-=sum[a-1][j][l];
if(ans<0)
{
ans+=mod;
}
}
return ans;
}
int main()
{
n=read();
k=read();
scanf("%s",s+1);
f[0][0][0]=f[0][1][0]=sum[0][0][0]=sum[0][1][0]=1;
int lb=-1,lw=-1;
for(int i=1; i<=n; ++i)
{
if(s[i]=='W')
{
lb=i;
}
else
{
int pb=std::max(lb,i-k+1);
f[i][1][0]=getsum(pb,i-1,0,0);
f[i][1][1]=getsum(lb,i-1,0,1);
if((pb>0)&&(pb<=i-k+1))
{
f[i][1][1]+=getsum(lb,pb-1,0,0);
if(f[i][1][1]>=mod)
{
f[i][1][1]-=mod;
}
}
f[i][1][2]=getsum(lb,i-1,0,2);
}
if(s[i]=='B')
{
lw=i;
}
else
{
int pw=std::max(lw,i-k+1);
f[i][0][0]=getsum(lw,i-1,1,0);
f[i][0][1]=getsum(pw,i-1,1,1);
f[i][0][2]=getsum(lw,i-1,1,2);
if((pw>0)&&(pw<=i-k+1))
{
f[i][0][2]+=getsum(lw,pw-1,1,1);
if(f[i][0][2]>=mod)
{
f[i][0][2]-=mod;
}
}
}
for(int j=0; j<2; ++j)
{
for(int l=0; l<3; ++l)
{
sum[i][j][l]=sum[i-1][j][l]+f[i][j][l];
if(sum[i][j][l]>=mod)
{
sum[i][j][l]-=mod;
}
}
}
}
int ans=f[n][0][2]+f[n][1][2];
if(ans>=mod)
{
ans-=mod;
}
printf("%d\n",ans);
return 0;
}