#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
int offset = 0;
struct zdrowy {
int dlugosc = 0;
int predkosc = 0;
int ratunek() {
int akt_dlugosc = dlugosc - offset * predkosc;
if (akt_dlugosc <= 0) return 0;
if (akt_dlugosc == 1) return 1;
return akt_dlugosc - predkosc + 1;
}
};
struct myclass {
bool operator() (zdrowy i, zdrowy j) { // czy powinien byc szybciej
if (i.predkosc == 1 && j.predkosc == 2 && i.ratunek() >= j.ratunek() - 1 && i.ratunek() > 0) return true;
if (j.predkosc == 1 && i.predkosc == 2 && j.ratunek() >= i.ratunek() - 1 && j.ratunek() > 0) return false;
return i.ratunek() == j.ratunek() ? i.predkosc < j.predkosc : i.ratunek() > j.ratunek();
}
} myobject;
int main() {
int t;
scanf("%d", &t);
for (int i = 0; i < t; i++) {
offset = 0;
vector<zdrowy> zdrowe;
//zdrowe.clear();
int n;
scanf("%d", &n);
getc(stdin);
int ile_jedynek = 0;
int ile_zer = 0;
bool czy_pierwsze_zero = false;
for (int j = 0; j < n; j++) {
char c = getc(stdin);
if (c == '1') {
ile_jedynek++;
if (ile_zer > 0) {
zdrowy zdr;
zdr.dlugosc = ile_zer;
zdr.predkosc = czy_pierwsze_zero && zdrowe.size() == 0 ? 1 : 2;
zdrowe.push_back(zdr);
ile_zer = 0;
}
} else {
ile_zer++;
if (j == 0) {
czy_pierwsze_zero = true;
}
if (j == n - 1) {
zdrowy zdr;
zdr.dlugosc = ile_zer;
zdr.predkosc = 1;
zdrowe.push_back(zdr);
}
}
}
if (ile_jedynek == 0 || ile_jedynek == n) {
printf("%d\n", ile_jedynek);
} else {
int result = 0;
vector<zdrowy>::iterator it = zdrowe.begin();
sort(it, zdrowe.end(), myobject);
while (it->ratunek() > 0 && it != zdrowe.end()) {
vector<zdrowy>::iterator it2;
// printf("-----\n");
// for (it2 = zdrowe.begin(); it2 != zdrowe.end(); ++it2) {
// printf("%d %d %d\n", it2->dlugosc, it2->predkosc, it2->ratunek());
// }
result += it->ratunek();
offset += it->predkosc;
it++;
sort(it, zdrowe.end(), myobject);
// printf("--- %d %d %d\n", it->dlugosc, it->predkosc, it->ratunek());
}
printf("%d\n", n - result);
}
}
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 | #include <cstdio> #include <vector> #include <algorithm> using namespace std; int offset = 0; struct zdrowy { int dlugosc = 0; int predkosc = 0; int ratunek() { int akt_dlugosc = dlugosc - offset * predkosc; if (akt_dlugosc <= 0) return 0; if (akt_dlugosc == 1) return 1; return akt_dlugosc - predkosc + 1; } }; struct myclass { bool operator() (zdrowy i, zdrowy j) { // czy powinien byc szybciej if (i.predkosc == 1 && j.predkosc == 2 && i.ratunek() >= j.ratunek() - 1 && i.ratunek() > 0) return true; if (j.predkosc == 1 && i.predkosc == 2 && j.ratunek() >= i.ratunek() - 1 && j.ratunek() > 0) return false; return i.ratunek() == j.ratunek() ? i.predkosc < j.predkosc : i.ratunek() > j.ratunek(); } } myobject; int main() { int t; scanf("%d", &t); for (int i = 0; i < t; i++) { offset = 0; vector<zdrowy> zdrowe; //zdrowe.clear(); int n; scanf("%d", &n); getc(stdin); int ile_jedynek = 0; int ile_zer = 0; bool czy_pierwsze_zero = false; for (int j = 0; j < n; j++) { char c = getc(stdin); if (c == '1') { ile_jedynek++; if (ile_zer > 0) { zdrowy zdr; zdr.dlugosc = ile_zer; zdr.predkosc = czy_pierwsze_zero && zdrowe.size() == 0 ? 1 : 2; zdrowe.push_back(zdr); ile_zer = 0; } } else { ile_zer++; if (j == 0) { czy_pierwsze_zero = true; } if (j == n - 1) { zdrowy zdr; zdr.dlugosc = ile_zer; zdr.predkosc = 1; zdrowe.push_back(zdr); } } } if (ile_jedynek == 0 || ile_jedynek == n) { printf("%d\n", ile_jedynek); } else { int result = 0; vector<zdrowy>::iterator it = zdrowe.begin(); sort(it, zdrowe.end(), myobject); while (it->ratunek() > 0 && it != zdrowe.end()) { vector<zdrowy>::iterator it2; // printf("-----\n"); // for (it2 = zdrowe.begin(); it2 != zdrowe.end(); ++it2) { // printf("%d %d %d\n", it2->dlugosc, it2->predkosc, it2->ratunek()); // } result += it->ratunek(); offset += it->predkosc; it++; sort(it, zdrowe.end(), myobject); // printf("--- %d %d %d\n", it->dlugosc, it->predkosc, it->ratunek()); } printf("%d\n", n - result); } } return 0; } |
English