glguy/aocpp
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

simpler

Browse Source
This commit is contained in:
Eric Mertens 2023-04-04 07:23:11 -07:00
parent c95b0b2daa
commit aef729afe4
1 changed files with 41 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

74
2022/14.cpp
View File

@ -5,7 +5,8 @@
#include <iostream>
#include <set>
#include <sstream>
#include <optional>
#include <stack>
#include <tuple>
#include <vector>
#include <boost/phoenix.hpp>
@ -16,7 +17,6 @@
#include <aocpp/Startup.hpp>
#include <aocpp/Coord.hpp>
#include <aocpp/Generator.hpp>
namespace {
@ -97,32 +97,33 @@ auto FindBottom(Input const& input) -> std::int64_t
return result;
}
auto SearchOrder(std::set<aocpp::Coord> & world, aocpp::Coord here) -> aocpp::Generator<aocpp::Coord>
{
co_yield aocpp::Down(here);
co_yield aocpp::Left(aocpp::Down(here));
co_yield aocpp::Right(aocpp::Down(here));
world.insert(here);
}
enum class Action { Visit, Mark };
auto Part1(std::int64_t bottom, std::set<aocpp::Coord> world) -> std::size_t
{
auto const starting_size = world.size();
auto work = std::stack<std::pair<Action, aocpp::Coord>>{};
work.emplace(Action::Visit, TOP);
auto path = std::vector<aocpp::Generator<aocpp::Coord>>{};
path.reserve(bottom - TOP.y);
path.push_back(SearchOrder(world, TOP));
while (!path.empty()) {
if (auto next = path.back()()) {
if (!world.contains(*next)) {
if (next->y == bottom) {
return world.size() - starting_size;
while (!work.empty()) {
auto const [action, here] = work.top();
work.pop();
switch (action) {
case Action::Visit:
if (!world.contains(here)) {
if (here.y == bottom) {
return world.size() - starting_size;
}
work.emplace(Action::Mark, here);
work.emplace(Action::Visit, aocpp::Right(aocpp::Down(here)));
work.emplace(Action::Visit, aocpp::Left(aocpp::Down(here)));
work.emplace(Action::Visit, aocpp::Down(here));
}
path.push_back(SearchOrder(world, *next));
}
} else {
path.pop_back();
break;
case Action::Mark:
world.insert(here);
break;
}
}
@ -132,18 +133,25 @@ auto Part1(std::int64_t bottom, std::set<aocpp::Coord> world) -> std::size_t
auto Part2(std::int64_t bottom, std::set<aocpp::Coord> world) -> std::size_t
{
auto const starting_size = world.size();
auto path = std::vector<aocpp::Generator<aocpp::Coord>>{};
path.reserve(2 + bottom - TOP.y);
path.push_back(SearchOrder(world, TOP));
auto work = std::stack<std::pair<Action, aocpp::Coord>>{};
work.emplace(Action::Visit, TOP);
while (!path.empty()) {
if (auto next = path.back()()) {
if (next->y - 2 != bottom && !world.contains(*next)) {
path.push_back(SearchOrder(world, *next));
}
} else {
path.pop_back();
while (!work.empty()) {
auto const [action, here] = work.top();
work.pop();
switch (action) {
case Action::Visit:
if (here.y - 2 != bottom && !world.contains(here)) {
work.emplace(Action::Mark, here);
work.emplace(Action::Visit, aocpp::Right(aocpp::Down(here)));
work.emplace(Action::Visit, aocpp::Left(aocpp::Down(here)));
work.emplace(Action::Visit, aocpp::Down(here));
}
break;
case Action::Mark:
world.insert(here);
break;
}
}