题解:枚举最左和最右发生改变的位置,可看作将j-i+1个位置上的1重新分配。如果末端位置的数为0,直接分配一个1,如果末端位置的数为1,分配一个零,然后将C[c1+c0][c0]加给答案;
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int mod=998244353;
int C[5005][5005],pre[5005];
int add(int x,int y){
return (x+y)%mod;}
int main()
{
ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
for(int i=0;i<=5000;i++)for(int j=0;j<=5000;j++)
{
if(i==0 || j==i)C[i][j]=1;
else C[i][j]=add(C[i-1][j-1],C[i-1][j]);
}
int n,k,ans=1;
cin>>n>>k;
string t;
cin>>t;
for(int i=0;i<n;i++)pre[i+1]=pre[i]+t[i]-'0';
for(int i=0;i<n;i++)
{
for(int j=i+1;j<n;j++)
{
int cnt=j-i+1;
int cnt1=pre[j+1]-pre[i];
if(pre[n]<k || cnt1>k)continue;
cnt-=2;
if(t[i]=='0')cnt1--;
if(t[j]=='0')cnt1--;
if(cnt>=cnt1 && cnt1>=0)
{
ans=add(ans,C[cnt][cnt1]);
}
}
}
cout<<ans<<endl;
return 0;
}