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
|
pub fn part1(input: &str) -> u32 {
let mut res: u32 = 0;
for (i, line) in input.trim_end().split('\n').enumerate() {
let game = line.split(": ").collect::<Vec<&str>>()[1];
let mut flag = true;
for point in game.split(";") {
for _play in point.split(',') {
let play = _play.trim();
let play_s: Vec<_> = play.split(' ').collect();
let (p, c) = (play_s[0].parse::<i32>().unwrap(), play_s[1].to_string());
if c.chars().nth(0).unwrap() == 'r' && p > 12 {
flag = false;
} else if c.chars().nth(0).unwrap() == 'g' && p > 13 {
flag = false;
} else if c.chars().nth(0).unwrap() == 'b' && p > 14 {
flag = false;
}
}
}
if flag {
res += (i + 1) as u32;
}
}
res
}
pub fn part2(input: &str) -> u32 {
let mut res: u32 = 0;
for line in input.trim_end().split('\n') {
let game = line.split(": ").collect::<Vec<&str>>()[1];
// r, g, b
let mut balls: [u32; 3] = [0, 0, 0];
for point in game.split(";") {
for _play in point.split(',') {
let play = _play.trim();
let play_s: Vec<_> = play.split(' ').collect();
let (p, c) = (play_s[0].parse::<u32>().unwrap(), play_s[1].to_string());
let bi = match c.chars().nth(0).unwrap() {
'r' => 0,
'g' => 1,
'b' => 2,
_ => panic!("wtf is this color?!"),
};
balls[bi] = if balls[bi] < p { p } else { balls[bi] };
}
}
println!("{:?}", balls);
let mut tot = 1;
for j in balls {
if j != 0 {
tot *= j;
}
}
res += tot;
}
res
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn example_part1() {
let input = include_str!("../example.txt");
let result = part1(input);
assert_eq!(result, 8);
}
#[test]
fn input_part1() {
let input = include_str!("../input.txt");
let result = part1(input);
assert_eq!(result, 2331);
}
#[test]
fn example_part2() {
let input = include_str!("../example.txt");
let result = part2(input);
assert_eq!(result, 2286);
}
#[test]
fn input_part2() {
let input = include_str!("../input.txt");
let result = part2(input);
assert_eq!(result, 71585);
}
}
|
