// B2 - Pandemia #include <iostream> #include <vector> #include <algorithm> using namespace std; struct Gap { int length; int sides; Gap(int l, int s) { length = l; sides = s; } bool operator<(const Gap& X) { return(length > X.length); } }; bool singleOverDouble(int s, int d) { return 2 * s >= d; } int main() { int t, n, prev, days, infected; char c; vector<Gap> gaps; int firstGap, secondGap; bool firstGapUsed, secondGapUsed; cin >> t; for (int i = 0; i < t; ++i) { cin >> n; prev = -1; infected = 0; for (int j = 0; j < n; ++j) { cin >> c; if (c == '1') { if (prev < 0) { firstGap = j; } else { if (j - prev > 1) { gaps.push_back(Gap( j - prev - 1, 2)); } } prev = j; ++infected; } } secondGap = n - 1 - prev; if (firstGap < secondGap) { firstGap = firstGap ^ secondGap; secondGap = firstGap ^ secondGap; firstGap = firstGap ^ secondGap; } if (infected == 0) { cout << 0 << endl; continue; } sort(gaps.begin(), gaps.end()); days = 0; firstGapUsed = false; secondGapUsed = false; for (unsigned j = 0; j < gaps.size(); ++j) { if (!firstGapUsed && singleOverDouble(firstGap - days, gaps[j].length - 2 * days)) { infected += min(days, firstGap); ++days; firstGapUsed = true; --j; } else if (!secondGapUsed && singleOverDouble(secondGap - days, gaps[j].length - 2 * days)) { infected += min(days, secondGap); ++days; secondGapUsed = true; --j; } else { if (gaps[j].length - days * 2 == 1) { infected += gaps[j].length - 1; days += 1; } else { infected += min(gaps[j].length, days * 2 + 1); days += 2; } } } if (!firstGapUsed) { infected += min(days, firstGap); ++days; firstGapUsed = true; } if (!secondGapUsed) { infected += min(days, secondGap); ++days; secondGapUsed = true; } cout << infected << endl; gaps.clear(); } 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 108 109 110 | // B2 - Pandemia #include <iostream> #include <vector> #include <algorithm> using namespace std; struct Gap { int length; int sides; Gap(int l, int s) { length = l; sides = s; } bool operator<(const Gap& X) { return(length > X.length); } }; bool singleOverDouble(int s, int d) { return 2 * s >= d; } int main() { int t, n, prev, days, infected; char c; vector<Gap> gaps; int firstGap, secondGap; bool firstGapUsed, secondGapUsed; cin >> t; for (int i = 0; i < t; ++i) { cin >> n; prev = -1; infected = 0; for (int j = 0; j < n; ++j) { cin >> c; if (c == '1') { if (prev < 0) { firstGap = j; } else { if (j - prev > 1) { gaps.push_back(Gap( j - prev - 1, 2)); } } prev = j; ++infected; } } secondGap = n - 1 - prev; if (firstGap < secondGap) { firstGap = firstGap ^ secondGap; secondGap = firstGap ^ secondGap; firstGap = firstGap ^ secondGap; } if (infected == 0) { cout << 0 << endl; continue; } sort(gaps.begin(), gaps.end()); days = 0; firstGapUsed = false; secondGapUsed = false; for (unsigned j = 0; j < gaps.size(); ++j) { if (!firstGapUsed && singleOverDouble(firstGap - days, gaps[j].length - 2 * days)) { infected += min(days, firstGap); ++days; firstGapUsed = true; --j; } else if (!secondGapUsed && singleOverDouble(secondGap - days, gaps[j].length - 2 * days)) { infected += min(days, secondGap); ++days; secondGapUsed = true; --j; } else { if (gaps[j].length - days * 2 == 1) { infected += gaps[j].length - 1; days += 1; } else { infected += min(gaps[j].length, days * 2 + 1); days += 2; } } } if (!firstGapUsed) { infected += min(days, firstGap); ++days; firstGapUsed = true; } if (!secondGapUsed) { infected += min(days, secondGap); ++days; secondGapUsed = true; } cout << infected << endl; gaps.clear(); } return 0; } |