diff options
| -rw-r--r-- | makefile | 3 | ||||
| -rw-r--r-- | src/day09.cpp | 107 |
2 files changed, 109 insertions, 1 deletions
| @@ -7,7 +7,8 @@ all: bin/day01 \ | |||
| 7 | bin/day05 \ | 7 | bin/day05 \ |
| 8 | bin/day06 \ | 8 | bin/day06 \ |
| 9 | bin/day07 \ | 9 | bin/day07 \ |
| 10 | bin/day08 | 10 | bin/day08 \ |
| 11 | bin/day09 | ||
| 11 | 12 | ||
| 12 | bin: | 13 | bin: |
| 13 | mkdir $@ | 14 | mkdir $@ |
diff --git a/src/day09.cpp b/src/day09.cpp new file mode 100644 index 0000000..1afcfe2 --- /dev/null +++ b/src/day09.cpp | |||
| @@ -0,0 +1,107 @@ | |||
| 1 | #include <fstream> | ||
| 2 | #include <iostream> | ||
| 3 | #include <sstream> | ||
| 4 | #include <string> | ||
| 5 | #include <vector> | ||
| 6 | using namespace std; | ||
| 7 | |||
| 8 | vector<string> | ||
| 9 | read_file(string_view filename) | ||
| 10 | { | ||
| 11 | fstream input{ filename }; | ||
| 12 | vector<string> data; | ||
| 13 | |||
| 14 | for ( string line; getline(input, line); ) { | ||
| 15 | data.emplace_back(line); | ||
| 16 | } | ||
| 17 | |||
| 18 | return data; | ||
| 19 | } | ||
| 20 | |||
| 21 | template<typename T = long> | ||
| 22 | vector<T> | ||
| 23 | read_ints(const string& line) | ||
| 24 | { | ||
| 25 | stringstream iss{ line }; | ||
| 26 | return vector<T>{ istream_iterator<T>{ iss }, istream_iterator<T>{} }; | ||
| 27 | } | ||
| 28 | |||
| 29 | template<typename T> | ||
| 30 | void | ||
| 31 | solve_rec_part1(vector<T>& values) | ||
| 32 | { | ||
| 33 | vector<T> differences; | ||
| 34 | |||
| 35 | for ( size_t idx = 1; idx < values.size(); ++idx ) { | ||
| 36 | auto difference = values[idx] - values[idx - 1]; | ||
| 37 | differences.emplace_back(difference); | ||
| 38 | } | ||
| 39 | auto all_zeros = all_of(differences.begin(), differences.end(), [](long diff) { return diff == 0; }); | ||
| 40 | if ( !all_zeros) { | ||
| 41 | solve_rec_part1(differences); | ||
| 42 | } | ||
| 43 | else { | ||
| 44 | differences.emplace_back(0); | ||
| 45 | } | ||
| 46 | auto last_value = values.back(); | ||
| 47 | auto last_difference = differences.back(); | ||
| 48 | values.emplace_back(last_value + last_difference); | ||
| 49 | } | ||
| 50 | |||
| 51 | void | ||
| 52 | part1() | ||
| 53 | { | ||
| 54 | auto lines = read_file("data/day09.txt"); | ||
| 55 | |||
| 56 | long sum = 0; | ||
| 57 | for ( const auto& line: lines ) { | ||
| 58 | auto values = read_ints(line); | ||
| 59 | solve_rec_part1(values); | ||
| 60 | sum += values.back(); | ||
| 61 | } | ||
| 62 | cout << sum << endl; | ||
| 63 | } | ||
| 64 | |||
| 65 | template<typename T> | ||
| 66 | void | ||
| 67 | solve_rec_part2(vector<T>& values) | ||
| 68 | { | ||
| 69 | vector<T> differences; | ||
| 70 | |||
| 71 | for ( size_t idx = 1; idx < values.size(); ++idx ) { | ||
| 72 | auto difference = values[idx] - values[idx - 1]; | ||
| 73 | differences.emplace_back(difference); | ||
| 74 | } | ||
| 75 | auto all_zeros = all_of(differences.begin(), differences.end(), [](long diff) { return diff == 0; }); | ||
| 76 | if ( !all_zeros) { | ||
| 77 | solve_rec_part2(differences); | ||
| 78 | } | ||
| 79 | else { | ||
| 80 | differences.insert(differences.begin(), 0); | ||
| 81 | } | ||
| 82 | |||
| 83 | auto first_value = values.front(); | ||
| 84 | auto first_difference = differences.front(); | ||
| 85 | values.insert(values.begin(), first_value - first_difference); | ||
| 86 | } | ||
| 87 | |||
| 88 | void | ||
| 89 | part2() | ||
| 90 | { | ||
| 91 | auto lines = read_file("data/day09.txt"); | ||
| 92 | |||
| 93 | long sum = 0; | ||
| 94 | for ( const auto& line: lines ) { | ||
| 95 | auto values = read_ints(line); | ||
| 96 | solve_rec_part2(values); | ||
| 97 | sum += values.front(); | ||
| 98 | } | ||
| 99 | cout << sum << endl; | ||
| 100 | } | ||
| 101 | |||
| 102 | int | ||
| 103 | main() | ||
| 104 | { | ||
| 105 | part1(); | ||
| 106 | part2(); | ||
| 107 | } | ||
