#include <bits/stdc++.h> using namespace std; #define REP(i,a,b) for (int i = (a); i <= (b); ++i) #define REPD(i,a,b) for (int i = (a); i >= (b); --i) #define FORI(i,n) REP(i,1,n) #define FOR(i,n) REP(i,0,int(n)-1) #define mp make_pair #define pb push_back #define pii pair<int,int> #define vi vector<int> #define ll long long #define SZ(x) int((x).size()) #define DBG(v) cerr << #v << " = " << (v) << endl; #define FOREACH(i,t) for (typeof(t.begin()) i=t.begin(); i!=t.end(); i++) #define fi first #define se second const int N = 100100; int n, m; char s[N]; int L[N], L0, L1; int suff[N]; void solve() { scanf("%d %s", &n, s); m = 0; L0 = L1 = -1; int add = 0; FOR(i,n) if (s[i] == '1') { if (L0 == -1) L0 = i; L1 = n-1-i; add++; } if (L0 == -1) { printf("0\n"); return; } int Lc = 0; for (int i = L0; i < n-L1; i++) { if (s[i] == '0') { Lc++; } else { if (Lc > 0) { L[m++] = -Lc; } Lc = 0; } } if (L0 < L1) swap(L0, L1); L[m++] = -1000000; sort(L,L+m); FOR(i,m) L[i] = -L[i]; //cerr << "L = " << L0 << " " << L1 << "\nL = "; //FOR(i,m) cerr << L[i] << " "; //cerr << "\n"; suff[m] = 0; for (int i = m-1; i > 0; i--) suff[i] = suff[i+1] + L[i]; int res = n-add; FOR(k,m) { // 1) t = 2*k int t = 2*k; if (L[k]>=t*2-1) { int sub = 0; if (k+1<m && L[k+1]>t*2) sub = 1; //cerr << k << " segments in time " << t << " : lowest segment taken = " << L[k] << ", suffix = " << suff[k+1] << "\t:\t " << t*(t-1)/2+L0+L1+suff[k+1] << "\n"; res = min(res, t*(t-1)/2+L0+L1+suff[k+1]-sub); } // 2) t = 2*k+1 t++; if (L[k]>=t*2-1 && L0>=t) { int sub = 0; if (k+1<m && L[k+1]>t*2) sub = 1; //cerr << k << " segments + one border in time " << t << " : lowest segment taken = " << L[k] << ", suffix = " << suff[k+1] << "\t:\t " << t*(t-1)/2+L1+suff[k+1] << "\n"; res = min(res, t*(t-1)/2+L1+suff[k+1]-sub); } // 3) t = 2*k+2 t++; if (L[k]>=t*2-1 && L1>=t) { int sub = 0; if (k+1<m && L[k+1]>t*2) sub = 1; //cerr << k << " segments + two borders in time " << t << " : lowest segment taken = " << L[k] << ", suffix = " << suff[k+1] << "\t:\t " << t*(t-1)/2+suff[k+1] << "\n"; res = min(res, t*(t-1)/2+suff[k+1]-sub); } } printf("%d\n", res+add); } int main() { int tt; scanf("%d", &tt); FOR(ii,tt) 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 | #include <bits/stdc++.h> using namespace std; #define REP(i,a,b) for (int i = (a); i <= (b); ++i) #define REPD(i,a,b) for (int i = (a); i >= (b); --i) #define FORI(i,n) REP(i,1,n) #define FOR(i,n) REP(i,0,int(n)-1) #define mp make_pair #define pb push_back #define pii pair<int,int> #define vi vector<int> #define ll long long #define SZ(x) int((x).size()) #define DBG(v) cerr << #v << " = " << (v) << endl; #define FOREACH(i,t) for (typeof(t.begin()) i=t.begin(); i!=t.end(); i++) #define fi first #define se second const int N = 100100; int n, m; char s[N]; int L[N], L0, L1; int suff[N]; void solve() { scanf("%d %s", &n, s); m = 0; L0 = L1 = -1; int add = 0; FOR(i,n) if (s[i] == '1') { if (L0 == -1) L0 = i; L1 = n-1-i; add++; } if (L0 == -1) { printf("0\n"); return; } int Lc = 0; for (int i = L0; i < n-L1; i++) { if (s[i] == '0') { Lc++; } else { if (Lc > 0) { L[m++] = -Lc; } Lc = 0; } } if (L0 < L1) swap(L0, L1); L[m++] = -1000000; sort(L,L+m); FOR(i,m) L[i] = -L[i]; //cerr << "L = " << L0 << " " << L1 << "\nL = "; //FOR(i,m) cerr << L[i] << " "; //cerr << "\n"; suff[m] = 0; for (int i = m-1; i > 0; i--) suff[i] = suff[i+1] + L[i]; int res = n-add; FOR(k,m) { // 1) t = 2*k int t = 2*k; if (L[k]>=t*2-1) { int sub = 0; if (k+1<m && L[k+1]>t*2) sub = 1; //cerr << k << " segments in time " << t << " : lowest segment taken = " << L[k] << ", suffix = " << suff[k+1] << "\t:\t " << t*(t-1)/2+L0+L1+suff[k+1] << "\n"; res = min(res, t*(t-1)/2+L0+L1+suff[k+1]-sub); } // 2) t = 2*k+1 t++; if (L[k]>=t*2-1 && L0>=t) { int sub = 0; if (k+1<m && L[k+1]>t*2) sub = 1; //cerr << k << " segments + one border in time " << t << " : lowest segment taken = " << L[k] << ", suffix = " << suff[k+1] << "\t:\t " << t*(t-1)/2+L1+suff[k+1] << "\n"; res = min(res, t*(t-1)/2+L1+suff[k+1]-sub); } // 3) t = 2*k+2 t++; if (L[k]>=t*2-1 && L1>=t) { int sub = 0; if (k+1<m && L[k+1]>t*2) sub = 1; //cerr << k << " segments + two borders in time " << t << " : lowest segment taken = " << L[k] << ", suffix = " << suff[k+1] << "\t:\t " << t*(t-1)/2+suff[k+1] << "\n"; res = min(res, t*(t-1)/2+suff[k+1]-sub); } } printf("%d\n", res+add); } int main() { int tt; scanf("%d", &tt); FOR(ii,tt) solve(); return 0; } |