#include <iostream>
#include <set>
enum Color : int {
YELLOW = 1,
BLUE = 2,
RED = 3,
};
struct Section {
int begin, end;
Color color;
friend std::istream &operator >> (std::istream &in, Section §ion) {
return in >> section.begin >> section.end >> (int &)section.color;
}
friend bool compareByBegins(const Section &s1, const Section &s2) {
return s1.begin < s2.begin || s1.begin == s2.begin && s1.end < s2.end || s1.begin == s2.begin && s1.end == s2.end && s1.color < s2.color;
}
friend bool compareByEnds(const Section &s1, const Section &s2) {
return s1.end < s2.end || s1.end == s2.end && s1.begin < s2.begin || s1.end == s2.end && s1.begin == s2.begin && s1.color < s2.color;
}
};
bool compareByBegins(const Section &s1, const Section &s2);
bool compareByEnds(const Section &s1, const Section &s2);
bool isGreen(const int levels[4]) {
return levels[RED] == 0 and levels[YELLOW] > 0 and levels[BLUE] > 0;
}
int main() {
std::ios_base::sync_with_stdio(false);
std::cin.tie(nullptr);
int n, m;
std::set<Section, decltype(compareByBegins)*> byBegins(compareByBegins);
std::set<Section, decltype(compareByEnds)*> byEnds(compareByEnds);
std::cin >> n >> m;
for (int i = 0; i < m; ++i) {
Section s{};
std::cin >> s;
byBegins.insert(s);
byEnds.insert(s);
}
int currentLevels[4] = {0};
int greenCans = 0, currentPos = 1;
auto beginsIter = byBegins.begin();
auto endsIter = byEnds.begin();
while (endsIter != byEnds.cend()) {
if (beginsIter != byBegins.cend() and beginsIter->begin <= endsIter->end) {
if (isGreen(currentLevels)) {
greenCans += beginsIter->begin - currentPos;
}
currentPos = beginsIter->begin;
++currentLevels[beginsIter->color];
++beginsIter;
} else {
if (isGreen(currentLevels)) {
if (endsIter->end - currentPos + 1) {
}
greenCans += endsIter->end - currentPos + 1;
}
currentPos = endsIter->end + 1;
--currentLevels[endsIter->color];
++endsIter;
}
}
std::cout << greenCans << std::endl;
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 | #include <iostream> #include <set> enum Color : int { YELLOW = 1, BLUE = 2, RED = 3, }; struct Section { int begin, end; Color color; friend std::istream &operator >> (std::istream &in, Section §ion) { return in >> section.begin >> section.end >> (int &)section.color; } friend bool compareByBegins(const Section &s1, const Section &s2) { return s1.begin < s2.begin || s1.begin == s2.begin && s1.end < s2.end || s1.begin == s2.begin && s1.end == s2.end && s1.color < s2.color; } friend bool compareByEnds(const Section &s1, const Section &s2) { return s1.end < s2.end || s1.end == s2.end && s1.begin < s2.begin || s1.end == s2.end && s1.begin == s2.begin && s1.color < s2.color; } }; bool compareByBegins(const Section &s1, const Section &s2); bool compareByEnds(const Section &s1, const Section &s2); bool isGreen(const int levels[4]) { return levels[RED] == 0 and levels[YELLOW] > 0 and levels[BLUE] > 0; } int main() { std::ios_base::sync_with_stdio(false); std::cin.tie(nullptr); int n, m; std::set<Section, decltype(compareByBegins)*> byBegins(compareByBegins); std::set<Section, decltype(compareByEnds)*> byEnds(compareByEnds); std::cin >> n >> m; for (int i = 0; i < m; ++i) { Section s{}; std::cin >> s; byBegins.insert(s); byEnds.insert(s); } int currentLevels[4] = {0}; int greenCans = 0, currentPos = 1; auto beginsIter = byBegins.begin(); auto endsIter = byEnds.begin(); while (endsIter != byEnds.cend()) { if (beginsIter != byBegins.cend() and beginsIter->begin <= endsIter->end) { if (isGreen(currentLevels)) { greenCans += beginsIter->begin - currentPos; } currentPos = beginsIter->begin; ++currentLevels[beginsIter->color]; ++beginsIter; } else { if (isGreen(currentLevels)) { if (endsIter->end - currentPos + 1) { } greenCans += endsIter->end - currentPos + 1; } currentPos = endsIter->end + 1; --currentLevels[endsIter->color]; ++endsIter; } } std::cout << greenCans << std::endl; return 0; } |
English