题目链接

https://www.lydsy.com/JudgeOnline/problem.php?id=2958

https://www.lydsy.com/JudgeOnline/problem.php?id=3269

双倍经验题,连题目名字都是一样的……

题解

f[i][0/1][0/1/2]f[i][0/1][0/1/2]表示前ii个位置,当前位置填’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;
}