#include <iostream> #include<algorithm> using namespace std; long long tab[1000001]; int main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); long long dlugosc_miasta, liczba_testow,zaraza=0,dzielnica=0, wartosc_miasto_l = 0, wartosc_miasto_p =0,dzien=0,x=0; bool skrajne_miasto_l = 0, skrajne_miasto_p = 0, zmienione_l = false, zmienione_p = false; string miasto; cin>>liczba_testow; for(int i=0;i<liczba_testow;i++) { cin>>dlugosc_miasta; cin>>miasto; if(miasto[0] == '1') { skrajne_miasto_l = true; } if(miasto[dlugosc_miasta-1] == '1' && dlugosc_miasta != 1) { skrajne_miasto_p = true; } for(int i=0; i<dlugosc_miasta; i++) { if(miasto[i] == '1') { zaraza++; if(tab[dzielnica] != 0) { dzielnica++; } } else { tab[dzielnica]=tab[dzielnica]+1; } } if(dzielnica ==0) cout<<zaraza<<endl; else { if(tab[dzielnica] == 0) dzielnica--; if(skrajne_miasto_l == false) { x=1; wartosc_miasto_l =tab[0]; } else { zmienione_l = true; } if(skrajne_miasto_p == false) { wartosc_miasto_p =tab[dzielnica]; dzielnica--; } else { zmienione_p = true; } sort(tab+x, tab+dzielnica+1); if(skrajne_miasto_l == false && skrajne_miasto_p == false) { wartosc_miasto_l=max(wartosc_miasto_l,wartosc_miasto_p); wartosc_miasto_p=min(wartosc_miasto_l,wartosc_miasto_p); } // for(int i=x;i<=dzielnica;i++) // { // cout<<tab[i]<<" "; // } // cout<<endl; for(int i=dzielnica; i>=x; i--) { if(wartosc_miasto_l-dzien > tab[i]-dzien*2 && zmienione_l == false) { if(wartosc_miasto_l-dzien>0)zaraza=zaraza+dzien; else zaraza=zaraza+wartosc_miasto_l; zmienione_l = true; dzien++; // cout<<"l: "<<zaraza<<endl; } if(wartosc_miasto_p-dzien > tab[i]-dzien*2 && zmienione_p == false) { if(wartosc_miasto_p-dzien>0)zaraza=zaraza+dzien; else zaraza=zaraza+wartosc_miasto_p; zmienione_p = true; dzien++; // cout<<"p: "<<zaraza<<endl; } if(tab[i]-dzien*2>0) { zaraza=zaraza+dzien*2; if(tab[i]-(dzien*2+1)>0) { zaraza++; dzien=dzien+2; } else { dzien++; } } else { zaraza=zaraza+tab[i]; dzien++; } // cout<<dzien<<" "<<zaraza<<endl; } if(zmienione_l == false) { if(wartosc_miasto_l-dzien>0)zaraza=zaraza+dzien; else zaraza=zaraza+wartosc_miasto_l; zmienione_l = true; dzien++; // cout<<"l: "<<zaraza<<endl; } if(zmienione_p == false) { if(wartosc_miasto_p-dzien>0)zaraza=zaraza+dzien; else zaraza=zaraza+wartosc_miasto_p; zmienione_p = true; dzien++; //cout<<"p: "<<zaraza<<endl; } cout<<zaraza<<endl; } for(int i=0; i<dlugosc_miasta; i++) { tab[i]=0; } zaraza=0,dzielnica=0, wartosc_miasto_l = 0, wartosc_miasto_p =0,dzien=0,x=0, skrajne_miasto_l = 0, skrajne_miasto_p = 0, zmienione_l = false, zmienione_p = false; } 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 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 | #include <iostream> #include<algorithm> using namespace std; long long tab[1000001]; int main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); long long dlugosc_miasta, liczba_testow,zaraza=0,dzielnica=0, wartosc_miasto_l = 0, wartosc_miasto_p =0,dzien=0,x=0; bool skrajne_miasto_l = 0, skrajne_miasto_p = 0, zmienione_l = false, zmienione_p = false; string miasto; cin>>liczba_testow; for(int i=0;i<liczba_testow;i++) { cin>>dlugosc_miasta; cin>>miasto; if(miasto[0] == '1') { skrajne_miasto_l = true; } if(miasto[dlugosc_miasta-1] == '1' && dlugosc_miasta != 1) { skrajne_miasto_p = true; } for(int i=0; i<dlugosc_miasta; i++) { if(miasto[i] == '1') { zaraza++; if(tab[dzielnica] != 0) { dzielnica++; } } else { tab[dzielnica]=tab[dzielnica]+1; } } if(dzielnica ==0) cout<<zaraza<<endl; else { if(tab[dzielnica] == 0) dzielnica--; if(skrajne_miasto_l == false) { x=1; wartosc_miasto_l =tab[0]; } else { zmienione_l = true; } if(skrajne_miasto_p == false) { wartosc_miasto_p =tab[dzielnica]; dzielnica--; } else { zmienione_p = true; } sort(tab+x, tab+dzielnica+1); if(skrajne_miasto_l == false && skrajne_miasto_p == false) { wartosc_miasto_l=max(wartosc_miasto_l,wartosc_miasto_p); wartosc_miasto_p=min(wartosc_miasto_l,wartosc_miasto_p); } // for(int i=x;i<=dzielnica;i++) // { // cout<<tab[i]<<" "; // } // cout<<endl; for(int i=dzielnica; i>=x; i--) { if(wartosc_miasto_l-dzien > tab[i]-dzien*2 && zmienione_l == false) { if(wartosc_miasto_l-dzien>0)zaraza=zaraza+dzien; else zaraza=zaraza+wartosc_miasto_l; zmienione_l = true; dzien++; // cout<<"l: "<<zaraza<<endl; } if(wartosc_miasto_p-dzien > tab[i]-dzien*2 && zmienione_p == false) { if(wartosc_miasto_p-dzien>0)zaraza=zaraza+dzien; else zaraza=zaraza+wartosc_miasto_p; zmienione_p = true; dzien++; // cout<<"p: "<<zaraza<<endl; } if(tab[i]-dzien*2>0) { zaraza=zaraza+dzien*2; if(tab[i]-(dzien*2+1)>0) { zaraza++; dzien=dzien+2; } else { dzien++; } } else { zaraza=zaraza+tab[i]; dzien++; } // cout<<dzien<<" "<<zaraza<<endl; } if(zmienione_l == false) { if(wartosc_miasto_l-dzien>0)zaraza=zaraza+dzien; else zaraza=zaraza+wartosc_miasto_l; zmienione_l = true; dzien++; // cout<<"l: "<<zaraza<<endl; } if(zmienione_p == false) { if(wartosc_miasto_p-dzien>0)zaraza=zaraza+dzien; else zaraza=zaraza+wartosc_miasto_p; zmienione_p = true; dzien++; //cout<<"p: "<<zaraza<<endl; } cout<<zaraza<<endl; } for(int i=0; i<dlugosc_miasta; i++) { tab[i]=0; } zaraza=0,dzielnica=0, wartosc_miasto_l = 0, wartosc_miasto_p =0,dzien=0,x=0, skrajne_miasto_l = 0, skrajne_miasto_p = 0, zmienione_l = false, zmienione_p = false; } return 0; } |