#include <iostream>
#include <vector>
using namespace std;
char zad[8002];
int zdal[8002];
int stac[8002];
int wolne[8002];
int n,k,t;
void licztab() {
int l1= 0, l2 = 0, l3 = 0;
for (int i=1; i<=n; i++) {
if (zad[i] == '1')
l1++;
if (zad[i] == '2')
l2++;
if (zad[i] == '3')
l3++;
stac[i] = l1;
zdal[i] = l2;
wolne[i] = l3;
}
}
int bez_podrozy() {
int lzdal=0, lstac=0, lwolne=0;
lzdal=zdal[n];
lstac=stac[n];
lwolne=wolne[n];
if (lstac <= k)
return(min(n, n-lstac-lzdal+k));
return(-1);
}
int z_podroza(int pocz, int kon) {
int lstac_o=stac[pocz+t-1]+stac[n]-stac[kon-1];
int lstac_o_doj=stac[pocz+t-1]-stac[pocz-1]+stac[kon+t-1]-stac[kon-1];
int lzdal_o_doj=zdal[pocz+t-1]-zdal[pocz-1]+zdal[kon+t-1]-zdal[kon-1];
int lzdal_o=zdal[pocz+t-1]-zdal[pocz-1]+zdal[kon+t-1]-zdal[kon-1];
int lwolne_o=wolne[kon+t-1]-wolne[pocz-1];
int w_pracy = stac[n] - lstac_o + zdal[kon-1] - zdal[pocz+t-1];
int zdal_dom = zdal[pocz-1] + zdal[n] - zdal[kon+t-1];
int stac_dom = stac[pocz-1] + stac[n] - stac[kon+t-1];
int wyn=-1;
if (lstac_o+lzdal_o > k)
return(-1);
wyn = n - lwolne_o - stac[n] - zdal[n] + min(zdal_dom+stac_dom, k-lstac_o_doj-lzdal_o_doj);
return(wyn);
}
int kwa() {
int wyn = -1;
int wynp = -1;
for (int i=1; i<=n-2*t; i++) {
for (int j=i+t; j<=n-t+1; j++) {
wynp = z_podroza(i,j);
if (wynp>wyn)
wyn=wynp;
}
}
return(wyn);
}
int main()
{
string line;
int wyn = -1;
int wynp = -1;
ios_base::sync_with_stdio(0);
cout.tie(0);
cin.tie(0);
cin >> n >> k >> t;
zad[0] = 0;
for (int i=1; i<=n; i++)
cin >> zad[i];
licztab();
wyn = bez_podrozy();
if (wyn >= 0) {
cout << wyn << endl;
return(0);
}
cout << kwa()<< endl;
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 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 | #include <iostream> #include <vector> using namespace std; char zad[8002]; int zdal[8002]; int stac[8002]; int wolne[8002]; int n,k,t; void licztab() { int l1= 0, l2 = 0, l3 = 0; for (int i=1; i<=n; i++) { if (zad[i] == '1') l1++; if (zad[i] == '2') l2++; if (zad[i] == '3') l3++; stac[i] = l1; zdal[i] = l2; wolne[i] = l3; } } int bez_podrozy() { int lzdal=0, lstac=0, lwolne=0; lzdal=zdal[n]; lstac=stac[n]; lwolne=wolne[n]; if (lstac <= k) return(min(n, n-lstac-lzdal+k)); return(-1); } int z_podroza(int pocz, int kon) { int lstac_o=stac[pocz+t-1]+stac[n]-stac[kon-1]; int lstac_o_doj=stac[pocz+t-1]-stac[pocz-1]+stac[kon+t-1]-stac[kon-1]; int lzdal_o_doj=zdal[pocz+t-1]-zdal[pocz-1]+zdal[kon+t-1]-zdal[kon-1]; int lzdal_o=zdal[pocz+t-1]-zdal[pocz-1]+zdal[kon+t-1]-zdal[kon-1]; int lwolne_o=wolne[kon+t-1]-wolne[pocz-1]; int w_pracy = stac[n] - lstac_o + zdal[kon-1] - zdal[pocz+t-1]; int zdal_dom = zdal[pocz-1] + zdal[n] - zdal[kon+t-1]; int stac_dom = stac[pocz-1] + stac[n] - stac[kon+t-1]; int wyn=-1; if (lstac_o+lzdal_o > k) return(-1); wyn = n - lwolne_o - stac[n] - zdal[n] + min(zdal_dom+stac_dom, k-lstac_o_doj-lzdal_o_doj); return(wyn); } int kwa() { int wyn = -1; int wynp = -1; for (int i=1; i<=n-2*t; i++) { for (int j=i+t; j<=n-t+1; j++) { wynp = z_podroza(i,j); if (wynp>wyn) wyn=wynp; } } return(wyn); } int main() { string line; int wyn = -1; int wynp = -1; ios_base::sync_with_stdio(0); cout.tie(0); cin.tie(0); cin >> n >> k >> t; zad[0] = 0; for (int i=1; i<=n; i++) cin >> zad[i]; licztab(); wyn = bez_podrozy(); if (wyn >= 0) { cout << wyn << endl; return(0); } cout << kwa()<< endl; return(0); } |
English