Line data    Source code

       1              : #include "PuzzleImpl.h"
       2              : #include "Regex.h"
       3              : 
       4              : #include <re2/re2.h>
       5              : 
       6              : #include <ranges>
       7              : 
       8              : namespace {
       9              : 
      10           23 : uint64_t parseMults(std::string_view const s) {
      11           23 :   static FindAndConsume<uint64_t, uint64_t> fac(R"(mul\((\d{1,3}),(\d{1,3})\))");
      12              : 
      13         1087 :   auto tupleProduct = [](auto const &t) { return get<0>(t) * get<1>(t); };
      14              : 
      15           23 :   return std::ranges::fold_left(fac(s) | std::views::transform(tupleProduct), uint64_t(0),
      16           23 :                                 std::plus{});
      17           23 : }
      18              : 
      19              : } // namespace
      20              : 
      21            1 : template <> std::string solvePart1<2024, 3>(std::string_view const input) {
      22            1 :   return std::to_string(parseMults(input));
      23            1 : }
      24              : 
      25            1 : template <> std::string solvePart2<2024, 3>(std::string_view const input) {
      26            1 :   std::string_view s = input;
      27            1 :   uint64_t result = 0;
      28              : 
      29            1 :   RE2::Options o;
      30            1 :   o.set_dot_nl(true);
      31            1 :   RE2 dontPattern(R"((.*?)don't\(\))", o);
      32            1 :   RE2 doPattern(R"(do\(\))");
      33              : 
      34            1 :   bool enabled = true;
      35           23 :   while (enabled) {
      36           22 :     std::string_view enabledMults;
      37           22 :     if (RE2::Consume(&s, dontPattern, &enabledMults)) {
      38           21 :       result += parseMults(enabledMults);
      39           21 :       enabled = RE2::FindAndConsume(&s, doPattern);
      40           21 :     } else {
      41            1 :       result += parseMults(s);
      42            1 :       enabled = false;
      43            1 :     }
      44           22 :   }
      45              : 
      46            1 :   return std::to_string(result);
      47            1 : }