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
|
use std::io::BufRead;
#[derive(Debug)]
struct Game {
id: usize,
red: u32,
blue: u32,
green: u32,
}
impl From<&str> for Game {
fn from(s: &str) -> Self {
let (game, sets) = s.split_once(':').unwrap();
let (_, id) = game.split_once(' ').unwrap();
let id = id.parse::<usize>().unwrap();
let (mut r, mut g, mut b) = (0, 0, 0);
let mut items: Vec<(u32, &str)> = vec![];
for set in sets.split(';') {
let sublist = set.split(',');
for list in sublist {
let list = list.trim_start();
let (num, color) = list.split_once(' ').unwrap();
items.push((num.parse::<u32>().unwrap(), color));
}
}
for (num, color) in items {
use std::cmp::max;
match color {
"red" => r = max(r, num),
"green" => g = max(g, num),
"blue" => b = max(b, num),
_ => {}
}
}
Self {
id,
red: r,
green: g,
blue: b,
}
}
}
fn solve_part_one(data: &str) -> u32 {
let mut games: Vec<Game> = vec![];
for game in data.lines() {
games.push(Game::from(game));
}
let mut ans: u32 = 0;
for game in games {
if game.red <= 12 && game.green <= 13 && game.blue <= 14 {
ans += game.id as u32;
}
}
ans
}
fn solve_part_two(data: &str) -> u32 {
let mut games: Vec<Game> = vec![];
for game in data.lines() {
games.push(Game::from(game));
}
games
.iter()
.map(|game| game.red * game.blue * game.green)
.sum::<u32>()
}
fn main() {
let test_1 = include_str!("../input/day2_1.test");
let prod = include_str!("../input/day2.prod");
println!("part_1 test: {:?}", solve_part_one(test_1));
println!("part_1 prod {:?}", solve_part_one(prod));
println!("part_2 test: {:?}", solve_part_two(test_1));
println!("part_2 prod {:?}", solve_part_two(prod));
}
|
