#include<bits/stdc++.h>
using namespace std;
int psum[8008][3];
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int ans=-1;
int n,k,t;
cin>>n>>k>>t;
string s;
cin>>s;
for(int i=1;i<=n;i++)
for(int j=0;j<3;j++)
psum[i][j]=psum[i-1][j]+(s[i-1]==(j+'1'));
if(psum[n][0]<=k)
{
int cur=psum[n][2];
int add=min(psum[n][1],k-psum[n][0]);
ans=max(ans,cur+add+psum[n][0]);
}
for(int i=1;i<=n;i++)
for(int j=i+t;j+t-1<=n;j++)
{
int msi=psum[i+t-1][0]+psum[i+t-1][1]-psum[i-1][0]-psum[i-1][1];
int msj=psum[j+t-1][0]+psum[j+t-1][1]-psum[j-1][0]-psum[j-1][1];
int mso=psum[i-1][0]+psum[n][0]-psum[j+t-1][0];
if(msi+msj+mso>k) continue;
int cur=psum[i-1][2]+psum[n][2]-psum[j+t-1][2];
int add=min(psum[i-1][1]+psum[n][1]-psum[j+t-1][1],k-msi-msj-mso);
ans=max(ans,cur+add+psum[i-1][0]+psum[n][0]-psum[j+t-1][0]);
}
cout<<ans<<'\n';
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | #include<bits/stdc++.h> using namespace std; int psum[8008][3]; int main() { ios::sync_with_stdio(false); cin.tie(0); cout.tie(0); int ans=-1; int n,k,t; cin>>n>>k>>t; string s; cin>>s; for(int i=1;i<=n;i++) for(int j=0;j<3;j++) psum[i][j]=psum[i-1][j]+(s[i-1]==(j+'1')); if(psum[n][0]<=k) { int cur=psum[n][2]; int add=min(psum[n][1],k-psum[n][0]); ans=max(ans,cur+add+psum[n][0]); } for(int i=1;i<=n;i++) for(int j=i+t;j+t-1<=n;j++) { int msi=psum[i+t-1][0]+psum[i+t-1][1]-psum[i-1][0]-psum[i-1][1]; int msj=psum[j+t-1][0]+psum[j+t-1][1]-psum[j-1][0]-psum[j-1][1]; int mso=psum[i-1][0]+psum[n][0]-psum[j+t-1][0]; if(msi+msj+mso>k) continue; int cur=psum[i-1][2]+psum[n][2]-psum[j+t-1][2]; int add=min(psum[i-1][1]+psum[n][1]-psum[j+t-1][1],k-msi-msj-mso); ans=max(ans,cur+add+psum[i-1][0]+psum[n][0]-psum[j+t-1][0]); } cout<<ans<<'\n'; return 0; } |
English