diff --git a/day15/part1.cpp b/day15/part1.cpp index fb53753..6c512f7 100644 --- a/day15/part1.cpp +++ b/day15/part1.cpp @@ -1,3 +1,4 @@ +#include #include #include #include @@ -7,8 +8,7 @@ constexpr long Y = 2000000; int main() { - std::set xs; - std::set bs; + std::bitset<8000000> xs; long px, py, bx, by; do { @@ -17,33 +17,24 @@ int main() if (std::cin.eof()) break; - auto a = sscanf(line.c_str(), - "Sensor at x=%ld, y=%ld: closest beacon is at x=%ld, y=%ld", - &px, &py, &bx, &by); + sscanf(line.c_str(), + "Sensor at x=%ld, y=%ld: closest beacon is at x=%ld, y=%ld", + &px, &py, &bx, &by); - std::cout << px << ' ' << py << ' ' << bx << ' ' << by; + long dist = std::abs(px - bx) + std::abs(py - by); - if (by == Y) - bs.insert(bx); - - long ic = std::abs(px - bx) + std::abs(py - by); - - if (Y < py - ic || Y > py + ic) + if (Y < py - dist || Y > py + dist) continue; - long spr = (ic - std::abs(py - Y)); + px += 1000000; + long spr = dist - std::abs(py - Y); for (long j = 0; j <= spr; ++j) { - xs.insert(px + j); - xs.insert(px - j); + xs.set(px + j); + xs.set(px - j); } - - std::cout << " -> " << xs.size() << std::endl; } while (1); - for (auto& b : bs) - xs.erase(b); - - std::cout << "Part 1: " << xs.size() << std::endl; + std::cout << "Part 1: " << xs.count() - 1 << std::endl; return 0; }