Remove some lines

This commit is contained in:
Simponic 2022-12-05 01:28:16 -07:00
parent eb419a7fbd
commit 4ed1b3307d
Signed by untrusted user who does not match committer: simponic
GPG Key ID: 52B3774857EB24B1
2 changed files with 26 additions and 34 deletions

BIN
day-04/a.out Executable file

Binary file not shown.

View File

@ -12,6 +12,12 @@
using stack_type = std::stack<char>; using stack_type = std::stack<char>;
enum PROBLEM
{
PROBLEM_1 = 1,
PROBLEM_2 = 2
};
std::tuple<std::vector<stack_type>, int> build_stacks(std::vector<std::string> &lines) std::tuple<std::vector<stack_type>, int> build_stacks(std::vector<std::string> &lines)
{ {
int num_stacks = std::ceil(lines[0].size() / 4.0); int num_stacks = std::ceil(lines[0].size() / 4.0);
@ -42,31 +48,7 @@ std::tuple<std::vector<stack_type>, int> build_stacks(std::vector<std::string> &
return std::make_tuple(stacks, lines_of_stacks); return std::make_tuple(stacks, lines_of_stacks);
} }
std::string solve1(std::vector<stack_type> &stacks, std::vector<std::string> &lines, int start_from) std::string solve(std::vector<stack_type> &stacks, std::vector<std::string> &lines, int start_from, PROBLEM problem)
{
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)
;
for (int i = 0; i < amount; i++)
{
stacks[to - 1].push(stacks[from - 1].top());
stacks[from - 1].pop();
}
}
std::string result;
for (auto &stack : stacks)
result += stack.top();
return result;
}
std::string solve2(std::vector<stack_type> &stacks, std::vector<std::string> &lines, int start_from)
{ {
stack_type curr; stack_type curr;
for (auto line = lines.begin() + start_from; line != lines.end(); ++line) for (auto line = lines.begin() + start_from; line != lines.end(); ++line)
@ -78,18 +60,28 @@ std::string solve2(std::vector<stack_type> &stacks, std::vector<std::string> &li
while (iss >> word >> amount >> word >> from >> word >> to) while (iss >> word >> amount >> word >> from >> word >> to)
; ;
for (int i = 0; i < amount; i++) 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()); curr.push(stacks[from - 1].top());
stacks[from - 1].pop(); stacks[from - 1].pop();
} }
for (int i = 0; i < amount; ++i)
for (int i = 0; i < amount; i++)
{ {
stacks[to - 1].push(curr.top()); stacks[to - 1].push(curr.top());
curr.pop(); curr.pop();
} }
} }
}
std::string result; std::string result;
for (auto &stack : stacks) for (auto &stack : stacks)
@ -105,9 +97,9 @@ int main()
lines.push_back(line); lines.push_back(line);
auto [stacks, start_from] = build_stacks(lines); auto [stacks, start_from] = build_stacks(lines);
std::cout << "Solve 1: " << solve1(stacks, lines, start_from + 1) << std::endl; std::cout << "Solve 1: " << solve(stacks, lines, start_from + 1, PROBLEM_1) << std::endl;
auto [stacks2, start_from2] = build_stacks(lines); auto [stacks2, start_from2] = build_stacks(lines);
std::cout << "Solve 2: " << solve2(stacks2, lines, start_from2 + 1) << std::endl; std::cout << "Solve 2: " << solve(stacks2, lines, start_from2 + 1, PROBLEM_2) << std::endl;
return 0; return 0;
} }