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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
use std::collections::HashMap;
use regex::Regex;
static DAY: u8 = 2;
fn main() {
let input = advent::read_lines(DAY);
println!("{DAY}a: {}", sum_possible(&input));
println!("{DAY}b: {}", sum_powers(&input));
}
struct Game {
id: usize,
infos: Vec<HashMap<String, usize>>,
}
impl Game {
fn new(input: &str) -> Game {
let re = Regex::new("^Game ([0-9]+): (.*)").unwrap();
let caps = re.captures(input).unwrap();
let id = caps[1].parse().unwrap();
let mut infos = Vec::new();
for info in caps[2].split("; ") {
let mut color_amount = HashMap::new();
for color_info in info.split(", ") {
let (amount, color) = color_info.split_once(' ').unwrap();
let amount = amount.parse().unwrap();
color_amount.insert(color.to_string(), amount);
}
infos.push(color_amount);
}
Game { id, infos }
}
fn is_possible(&self, bagged: &HashMap<String, usize>) -> bool {
for info in &self.infos {
for (color, amount) in info {
if bagged[color] < *amount {
return false
}
}
}
true
}
fn power(&self) -> usize {
let mut least_amount = HashMap::new();
for info in &self.infos {
for (color, amount) in info {
let least = least_amount.entry(color).or_insert(0);
if amount > least {
least_amount.insert(color, *amount);
}
}
}
least_amount.values()
.product()
}
}
fn sum_possible(input: &[String]) -> usize {
let games = input.iter()
.map(|x| Game::new(x))
.collect::<Vec<Game>>();
let bag_content = HashMap::from([
("red".to_string(), 12),
("green".to_string(), 13),
("blue".to_string(), 14),
]);
games.iter()
.filter(|x| x.is_possible(&bag_content))
.map(|x| x.id)
.sum()
}
fn sum_powers(input: &[String]) -> usize {
input.iter()
.map(|x| Game::new(x))
.map(|g| g.power())
.sum()
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test() {
let input = [
"Game 1: 3 blue, 4 red; 1 red, 2 green, 6 blue; 2 green",
"Game 2: 1 blue, 2 green; 3 green, 4 blue, 1 red; 1 green, 1 blue",
"Game 3: 8 green, 6 blue, 20 red; 5 blue, 4 red, 13 green; 5 green, 1 red",
"Game 4: 1 green, 3 red, 6 blue; 3 green, 6 red; 3 green, 15 blue, 14 red",
"Game 5: 6 red, 1 blue, 3 green; 2 blue, 1 red, 2 green",
].iter().map(|&x| String::from(x)).collect::<Vec<_>>();
assert_eq!(sum_possible(&input), 8);
assert_eq!(sum_powers(&input), 2286);
}
}
|
