解题思路
题目大意:给定一个字符串修改不超过k次且不含有110子串的方案数
思路:可以考虑动态规划,动态规划状态表示:f[i][j][k]表示前i个字符修改不超过j次且后三个字符组成的二进制值为k的所有不含有110子串的方案数,状态转移可知有8个状态
不修改第i个字符
if(a[i]==0)
f[i][j][0]=(f[i-1][j][0]+f[i-1][j][4])%mod;
f[i][j][2]=(f[i-1][j][1]+f[i-1][j][5])%mod;
f[i][j][4]=(f[i-1][j][2])%mod;
f[i][j][6]=0;
else
f[i][j][1]=(f[i-1][j][0]+f[i-1][j][4])%mod;
f[i][j][3]=(f[i-1][j][1]+f[i-1][j][5])%mod;
f[i][j][5]=(f[i-1][j][2])%mod;
f[i][j][7]=(f[i-1][j][3]+f[i-1][j][7])%mod;
修改第i个字符
if(a[i]==0)
f[i][j][1]=(f[i-1][j-1][0]+f[i-1][j-1][4])%mod;
f[i][j][3]=(f[i-1][j-1][1]+f[i-1][j-1][5])%mod;
f[i][j][5]=(f[i-1][j-1][2])%mod;
f[i][j][7]=(f[i-1][j-1][3]+f[i-1][j-1][7])%mod;
else
f[i][j][0]=(f[i-1][j-1][0]+f[i-1][j-1][4])%mod;
f[i][j][2]=(f[i-1][j-1][1]+f[i-1][j-1][5])%mod;
f[i][j][4]=(f[i-1][j-1][2])%mod;
f[i][j-1][6]=0;
动态规划的核心:状态表示要不重不漏,状态转移要正确的转移过来
我们计算一下所开的空间5e3*5e3*8*4*B=800MB>64MB所以超内存了,时间复杂度为8*5e3*5e3=2*108应该刚刚好能过,内存超出了应该怎么办呢发现当前状态都是由前一个状态转化过来的可以用f[0][j][k]表示前一个状态f[1][j][k]表示当前的状态这样内存为2*8*5e3*4B=32e4B不超内存
代码
#include<iostream>
using namespace std;
const int N=5e3+5,mod=998244353;
int f[2][N][8];
char a[N];
int n,k;
signed main()
{
cin>>n>>k;
cin>>a+1;
f[0][0][0]=1;
for(int i=1;i<=n;i++)
{
for(int j=0;j<=k;j++)
{
if(a[i]=='0')
{
f[1][j][0]=(f[0][j][0]+f[0][j][4])%mod;
f[1][j][2]=(f[0][j][1]+f[0][j][5])%mod;
f[1][j][4]=(f[0][j][2])%mod;
f[1][j][6]=0;
if(j)
{
f[1][j][1]=(f[0][j-1][0]+f[0][j-1][4])%mod;
f[1][j][3]=(f[0][j-1][1]+f[0][j-1][5])%mod;
f[1][j][5]=(f[0][j-1][2])%mod;
f[1][j][7]=(f[0][j-1][3]+f[0][j-1][7])%mod;
}
}
else
{
f[1][j][1]=(f[0][j][0]+f[0][j][4])%mod;
f[1][j][3]=(f[0][j][1]+f[0][j][5])%mod;
f[1][j][5]=(f[0][j][2])%mod;
f[1][j][7]=(f[0][j][3]+f[0][j][7])%mod;
if(j)
{
f[1][j][0]=(f[0][j-1][0]+f[0][j-1][4])%mod;
f[1][j][2]=(f[0][j-1][1]+f[0][j-1][5])%mod;
f[1][j][4]=(f[0][j-1][2])%mod;
f[1][j-1][6]=0;}
}
}
for(int j=0;j<=k;j++)
{
for(int k=0;k<=7;k++)
{
f[0][j][k]=f[1][j][k];
f[1][j][k]=0;
}
}
}
int ans=0;
for(int i=0;i<=k;i++)
{
for(int j=0;j<=7;j++)
ans=(ans+f[0][i][j])%mod;
}
cout<<ans%mod;
return 0;
}