#include <iostream> #include <vector> #include <algorithm> #include <set> #include <string> using namespace std; int n,k,t; string s; // length, borders, (last_update), set <pair<int,int>> v1; set <pair<int,int>> v2; void readData(){ v1.clear(); v2.clear(); scanf("%d", &n); cin>>s; // cout<<s; int zeros = 0; bool first = true; for (int i =0; i < s.size(); i++){ if (s[i] == '0'){ zeros ++; }else{ if (first){ if ( zeros > 0) v1.insert(make_pair((-1)*zeros,-i)); zeros = 0; first = false; }else{ if ( zeros > 0) v2.insert(make_pair((-1)*zeros, i)); zeros = 0; } } } if ( zeros > 0 ){ v1.insert(make_pair((-1)*zeros,(s.size()-1)*(-1) )); } } void print_v1 (){ auto it = v1.begin(); for (; it != v1.end(); ++it) { pair<int,int> p = *it; // Note the "*" here cout<<p.first << " "<< p.second <<"\n"; } } void print_v2 (){ auto it = v2.begin(); for (; it != v2.end(); ++it) { pair<int,int> p = *it; // Note the "*" here cout<<p.first << " "<< p.second <<"\n"; } } int getAns(){ int time = 0; long long score = 0; for ( ; v1.begin()!=v1.end() || v2.begin()!=v2.end(); time++){ // cout<<"TIME : "<<time<<"score : "<<score<<"\n"<<"v1:\n"; // print_v1(); // cout<<"v2:\n"; // print_v2(); if (v1.begin()==v1.end()){ pair<int,int> p = *v2.begin(); v2.erase(v2.begin()); p.first+=time; if ( p.first < 0 ){ v1.insert(p); } continue; } //v1 not empty pair<int,int> p1 = *v1.begin(); if (v2.begin()!=v2.end()){ pair<int,int> p2 = *v2.begin(); if ( p2.first * (-1) > p1.first * (-1) + 1){ v2.erase(v2.begin()); p2.first += time; if ( p2.first < 0 ){ v1.insert(p2); } continue; } else{ v1.erase(v1.begin()); p1.first+=time; if (p1.first < 0 ){ score += (-1) * p1.first; } else{ if (p1.second >= 0) score++; } } } else{ v1.erase(v1.begin()); p1.first+=time; if (p1.first < 0 ){ score += (-1) * p1.first; } else { if (p1.second >= 0) score++; } } } return n-score; } int main(){ scanf("%d", &t); for (int i = 0; i< t; i++){ readData(); int ans = getAns(); printf("%d\n", ans ); } }
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 | #include <iostream> #include <vector> #include <algorithm> #include <set> #include <string> using namespace std; int n,k,t; string s; // length, borders, (last_update), set <pair<int,int>> v1; set <pair<int,int>> v2; void readData(){ v1.clear(); v2.clear(); scanf("%d", &n); cin>>s; // cout<<s; int zeros = 0; bool first = true; for (int i =0; i < s.size(); i++){ if (s[i] == '0'){ zeros ++; }else{ if (first){ if ( zeros > 0) v1.insert(make_pair((-1)*zeros,-i)); zeros = 0; first = false; }else{ if ( zeros > 0) v2.insert(make_pair((-1)*zeros, i)); zeros = 0; } } } if ( zeros > 0 ){ v1.insert(make_pair((-1)*zeros,(s.size()-1)*(-1) )); } } void print_v1 (){ auto it = v1.begin(); for (; it != v1.end(); ++it) { pair<int,int> p = *it; // Note the "*" here cout<<p.first << " "<< p.second <<"\n"; } } void print_v2 (){ auto it = v2.begin(); for (; it != v2.end(); ++it) { pair<int,int> p = *it; // Note the "*" here cout<<p.first << " "<< p.second <<"\n"; } } int getAns(){ int time = 0; long long score = 0; for ( ; v1.begin()!=v1.end() || v2.begin()!=v2.end(); time++){ // cout<<"TIME : "<<time<<"score : "<<score<<"\n"<<"v1:\n"; // print_v1(); // cout<<"v2:\n"; // print_v2(); if (v1.begin()==v1.end()){ pair<int,int> p = *v2.begin(); v2.erase(v2.begin()); p.first+=time; if ( p.first < 0 ){ v1.insert(p); } continue; } //v1 not empty pair<int,int> p1 = *v1.begin(); if (v2.begin()!=v2.end()){ pair<int,int> p2 = *v2.begin(); if ( p2.first * (-1) > p1.first * (-1) + 1){ v2.erase(v2.begin()); p2.first += time; if ( p2.first < 0 ){ v1.insert(p2); } continue; } else{ v1.erase(v1.begin()); p1.first+=time; if (p1.first < 0 ){ score += (-1) * p1.first; } else{ if (p1.second >= 0) score++; } } } else{ v1.erase(v1.begin()); p1.first+=time; if (p1.first < 0 ){ score += (-1) * p1.first; } else { if (p1.second >= 0) score++; } } } return n-score; } int main(){ scanf("%d", &t); for (int i = 0; i< t; i++){ readData(); int ans = getAns(); printf("%d\n", ans ); } } |