From 3d57434c04a669610d5f15bd2a7713e6928cdef7 Mon Sep 17 00:00:00 2001 From: Elizabeth Hunt Date: Thu, 30 Nov 2023 22:46:45 -0700 Subject: add aoc2023 --- aoc_2022/day-04/sol.cpp | 103 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 103 insertions(+) create mode 100644 aoc_2022/day-04/sol.cpp (limited to 'aoc_2022/day-04/sol.cpp') diff --git a/aoc_2022/day-04/sol.cpp b/aoc_2022/day-04/sol.cpp new file mode 100644 index 0000000..d7847da --- /dev/null +++ b/aoc_2022/day-04/sol.cpp @@ -0,0 +1,103 @@ +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +using stack_type = std::stack; + +enum PROBLEM +{ + PROBLEM_1 = 1, + PROBLEM_2 = 2 +}; + +std::tuple, int> build_stacks(std::vector &lines) +{ + int num_stacks = std::ceil(lines[0].size() / 4.0); + + std::vector stacks(num_stacks); + + std::vector stack_positions(num_stacks, 0); + int lines_of_stacks = 0; + for (std::string line : lines) + { + char l = '1'; + for (int i = 0; i < line.size(); ++i) + if (line[i] == l) + { + stack_positions[l - '1'] = i; + l++; + } + if (l == num_stacks + '1') + break; + lines_of_stacks++; + } + + for (int i = lines_of_stacks - 1; i >= 0; i--) + for (int j = 0; j < num_stacks; ++j) + if (lines[i][stack_positions[j]] != ' ') + stacks[j].push(lines[i][stack_positions[j]]); + + return std::make_tuple(stacks, lines_of_stacks); +} + +std::string solve(std::vector &stacks, std::vector &lines, int start_from, PROBLEM problem) +{ + stack_type curr; + for (auto line = lines.begin() + start_from; line != lines.end(); ++line) + { + std::istringstream iss(*line); + std::string word; + int from = 0, to = 0, amount = 0; + + while (iss >> word >> amount >> word >> from >> word >> to) + ; + + if (problem == PROBLEM_1) + for (int i = 0; i < amount; ++i) + { + stacks[to - 1].push(stacks[from - 1].top()); + stacks[from - 1].pop(); + } + else + { + for (int i = 0; i < amount; ++i) + { + curr.push(stacks[from - 1].top()); + stacks[from - 1].pop(); + } + for (int i = 0; i < amount; ++i) + { + stacks[to - 1].push(curr.top()); + curr.pop(); + } + } + } + + std::string result; + for (auto &stack : stacks) + result += stack.top(); + return result; +} + +int main() +{ + std::ifstream input("input"); + std::vector lines; + for (std::string line; std::getline(input, line);) + lines.push_back(line); + + auto [stacks, start_from] = build_stacks(lines); + std::cout << "Solve 1: " << solve(stacks, lines, start_from + 1, PROBLEM_1) << std::endl; + auto [stacks2, start_from2] = build_stacks(lines); + std::cout << "Solve 2: " << solve(stacks2, lines, start_from2 + 1, PROBLEM_2) << std::endl; + + return 0; +} -- cgit v1.2.3-70-g09d2