#include<bits/stdc++.h> #define fi first #define se second #define pb push_back using namespace std; vector<int> v; string s; void solve() { int n, pocz = 0, kon = 0, a = 0, b = 0, wyn = 0; cin>>n>>s; v.clear(); while(s[a] == '0' && a < s.size()) { pocz++; a++; } for(int i=a; i<s.size(); i++) { if(s[i] == '1') { if(b > 0) { v.pb(b); b = 0; } wyn++; } else b++; } kon = b; sort(v.begin(), v.end(), greater<int>()); a = 0; for(int i=0; i<v.size(); i++) { b = v[i]; if(pocz > 0 && 2 * pocz - 1 >= b) { if(pocz > a) { wyn += a; a++; } else wyn += pocz; pocz = -1; } if(kon > 0 && 2 * kon - 1 >= b) { if(kon > a) { wyn += a; a++; } else wyn += kon; kon = -1; } if(2 * a + 1 < b) { wyn += (2 * a + 1); a += 2; } else if(2 * a + 1 == b) { wyn += 2 * a; a++; } else wyn += b; } if(pocz > 0) { if(pocz > a) { wyn += a; a++; } else wyn += pocz; } if(kon > 0) { if(kon > a) { wyn += a; a++; } else wyn += kon; } cout<<wyn<<"\n"; return; } int main() { ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0); int t; cin>>t; while(t--) solve(); 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 | #include<bits/stdc++.h> #define fi first #define se second #define pb push_back using namespace std; vector<int> v; string s; void solve() { int n, pocz = 0, kon = 0, a = 0, b = 0, wyn = 0; cin>>n>>s; v.clear(); while(s[a] == '0' && a < s.size()) { pocz++; a++; } for(int i=a; i<s.size(); i++) { if(s[i] == '1') { if(b > 0) { v.pb(b); b = 0; } wyn++; } else b++; } kon = b; sort(v.begin(), v.end(), greater<int>()); a = 0; for(int i=0; i<v.size(); i++) { b = v[i]; if(pocz > 0 && 2 * pocz - 1 >= b) { if(pocz > a) { wyn += a; a++; } else wyn += pocz; pocz = -1; } if(kon > 0 && 2 * kon - 1 >= b) { if(kon > a) { wyn += a; a++; } else wyn += kon; kon = -1; } if(2 * a + 1 < b) { wyn += (2 * a + 1); a += 2; } else if(2 * a + 1 == b) { wyn += 2 * a; a++; } else wyn += b; } if(pocz > 0) { if(pocz > a) { wyn += a; a++; } else wyn += pocz; } if(kon > 0) { if(kon > a) { wyn += a; a++; } else wyn += kon; } cout<<wyn<<"\n"; return; } int main() { ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0); int t; cin>>t; while(t--) solve(); return 0; } |