1use crate::util::parse::*;
6
7type Input = (u32, u32);
8
9pub fn parse(input: &str) -> Input {
10 let memory = input.as_bytes();
11 let mut index = 0;
12 let mut enabled = true;
13 let mut part_one = 0;
14 let mut part_two = 0;
15
16 while index < memory.len() {
17 if memory[index] != b'm' && memory[index] != b'd' {
19 index += 1;
20 continue;
21 }
22
23 if memory[index..].starts_with(b"mul(") {
25 index += 4;
26 } else if memory[index..].starts_with(b"do()") {
27 index += 4;
28 enabled = true;
29 continue;
30 } else if memory[index..].starts_with(b"don't()") {
31 index += 7;
32 enabled = false;
33 continue;
34 } else {
35 index += 1;
36 continue;
37 }
38
39 let first = parse_number(memory, &mut index);
41
42 if memory[index] != b',' {
44 continue;
45 }
46 index += 1;
47
48 let second = parse_number(memory, &mut index);
50
51 if memory[index] != b')' {
53 continue;
54 }
55 index += 1;
56
57 let product = first * second;
59 part_one += product;
60 part_two += if enabled { product } else { 0 };
61 }
62
63 (part_one, part_two)
64}
65
66fn parse_number(memory: &[u8], index: &mut usize) -> u32 {
67 let mut number = 0;
68 while memory[*index].is_ascii_digit() {
69 number = 10 * number + memory[*index].to_decimal() as u32;
70 *index += 1;
71 }
72 number
73}
74
75pub fn part1(input: &Input) -> u32 {
76 input.0
77}
78
79pub fn part2(input: &Input) -> u32 {
80 input.1
81}