#include <bits/stdc++.h> using namespace std; int t, n, c, z; string str; multiset<int, greater<> > s1, s2; int main() { cin.tie(0); cout.tie(0); ios_base::sync_with_stdio(0); cin>>t; while(t--) { cin>>n>>str; c=0; for(int i=0; i<n; ++i) { if(str[i]=='1') { if(c) { if(str[0]=='0'&&s1.empty())s1.insert(c); else s2.insert(c); c=0; } } else ++c; } if(c) { s1.insert(c); c=0; } z=0; while(!s1.empty()||!s2.empty()) { if(!s1.empty()&&!s2.empty()) { if((*s1.begin()-c>=1&&(*s2.begin()-2*c==*s1.begin()-c+1||*s2.begin()-2*c==*s1.begin()-c+2))||(*s1.begin()-c)>=(*s2.begin()-2*c)) { z+=max((*s1.begin()-c), 0); s1.erase(s1.begin()); } else { s1.insert(*s2.begin()-c-1); if(*s2.begin()-2*c>0)++z; s2.erase(s2.begin()); } } else if(!s1.empty()) { z+=max((*s1.begin()-c), 0); s1.erase(s1.begin()); } else { s1.insert(*s2.begin()-c-1); if(*s2.begin()-2*c>0)++z; s2.erase(s2.begin()); } ++c; } cout<<n-z<<"\n"; s1=s2=multiset<int, greater<> >(); } }
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 | #include <bits/stdc++.h> using namespace std; int t, n, c, z; string str; multiset<int, greater<> > s1, s2; int main() { cin.tie(0); cout.tie(0); ios_base::sync_with_stdio(0); cin>>t; while(t--) { cin>>n>>str; c=0; for(int i=0; i<n; ++i) { if(str[i]=='1') { if(c) { if(str[0]=='0'&&s1.empty())s1.insert(c); else s2.insert(c); c=0; } } else ++c; } if(c) { s1.insert(c); c=0; } z=0; while(!s1.empty()||!s2.empty()) { if(!s1.empty()&&!s2.empty()) { if((*s1.begin()-c>=1&&(*s2.begin()-2*c==*s1.begin()-c+1||*s2.begin()-2*c==*s1.begin()-c+2))||(*s1.begin()-c)>=(*s2.begin()-2*c)) { z+=max((*s1.begin()-c), 0); s1.erase(s1.begin()); } else { s1.insert(*s2.begin()-c-1); if(*s2.begin()-2*c>0)++z; s2.erase(s2.begin()); } } else if(!s1.empty()) { z+=max((*s1.begin()-c), 0); s1.erase(s1.begin()); } else { s1.insert(*s2.begin()-c-1); if(*s2.begin()-2*c>0)++z; s2.erase(s2.begin()); } ++c; } cout<<n-z<<"\n"; s1=s2=multiset<int, greater<> >(); } } |