blob: caa2acf3fa5550bca43badccc25ed636a2291d31 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
#include "aoc.h"
#include <algorithm>
namespace aoc2020 {
bool find_color(color_bag* bag, line_view color) {
if (bag->color == color) {
return true;
} else {
for (auto& p : bag->bags) {
if (find_color(p.first, color)) {
return true;
}
}
return false;
}
}
void count_color(color_bag* bag, int* total) {
for (auto& b : bag->bags) {
*total += b.second;
for (int i = 0; i < b.second; i++) {
count_color(b.first, total);
}
}
}
std::pair<int, int> day7(line_view file, const char* color) {
bag_regulations rs;
per_line(file, [&rs](line_view lv) {
rs.parse(lv);
return true;
});
int t0{0};
for (auto& b : rs.regulations) {
if (!(b.second->color == color)) {
t0 += int(find_color(b.second, color));
}
}
int t1{0};
count_color(rs.regulations[color], &t1);
return {t0, t1};
}
} // namespace aoc2020
|
