diff options
Diffstat (limited to '2023/rust/day6')
-rw-r--r-- | 2023/rust/day6/Cargo.toml | 8 | ||||
-rw-r--r-- | 2023/rust/day6/example.txt | 2 | ||||
-rw-r--r-- | 2023/rust/day6/input.txt | 3 | ||||
-rw-r--r-- | 2023/rust/day6/src/lib.rs | 102 |
4 files changed, 115 insertions, 0 deletions
diff --git a/2023/rust/day6/Cargo.toml b/2023/rust/day6/Cargo.toml new file mode 100644 index 0000000..89d04ae --- /dev/null +++ b/2023/rust/day6/Cargo.toml @@ -0,0 +1,8 @@ +[package] +name = "day6" +version = "0.1.0" +edition = "2021" + +# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html + +[dependencies] diff --git a/2023/rust/day6/example.txt b/2023/rust/day6/example.txt new file mode 100644 index 0000000..28f5ae9 --- /dev/null +++ b/2023/rust/day6/example.txt @@ -0,0 +1,2 @@ +Time: 7 15 30 +Distance: 9 40 200 diff --git a/2023/rust/day6/input.txt b/2023/rust/day6/input.txt new file mode 100644 index 0000000..f4a8fdc --- /dev/null +++ b/2023/rust/day6/input.txt @@ -0,0 +1,3 @@ +Time: 41 66 72 66 +Distance: 244 1047 1228 1040 + diff --git a/2023/rust/day6/src/lib.rs b/2023/rust/day6/src/lib.rs new file mode 100644 index 0000000..b3e33cb --- /dev/null +++ b/2023/rust/day6/src/lib.rs @@ -0,0 +1,102 @@ +pub fn part1(input: &str) -> u64 { + let mut res: u64 = 1; + + let td: Vec<Vec<_>> = input + .trim_end() + .split('\n') + .map(|x| { + x.split(':') + .nth(1) + .unwrap() + .trim() + .replace(" ", " ") + .replace(" ", " ") + .replace(" ", " ") + .split(' ') + .map(|y| y.parse::<u64>().unwrap()) + .collect() + }) + .collect(); + + assert_eq!(td.len(), 2); + let (t, d) = (&td[0], &td[1]); + + for i in 0..t.len() { + let mut r = 0; + for j in 0..t[i] { + if (t[i] - j) * j > d[i] { + r += 1; + } + } + res *= r; + } + + res +} + +pub fn part2(input: &str) -> u64 { + let mut res: u64 = 1; + + let td: Vec<Vec<_>> = input + .trim_end() + .split('\n') + .map(|x| { + x.split(':') + .nth(1) + .unwrap() + .trim() + .replace(" ", "") + .split(' ') + .map(|y| y.parse::<u64>().unwrap()) + .collect() + }) + .collect(); + + assert_eq!(td.len(), 2); + let (t, d) = (&td[0], &td[1]); + + for i in 0..t.len() { + let mut r = 0; + for j in 0..t[i] { + if (t[i] - j) * j > d[i] { + r += 1; + } + } + res *= r; + } + + res +} + +#[cfg(test)] +mod tests { + use super::*; + + #[test] + fn example_part1() { + let input = include_str!("../example.txt"); + let result = part1(input); + assert_eq!(result, 288); + } + + #[test] + fn input_part1() { + let input = include_str!("../input.txt"); + let result = part1(input); + assert_eq!(result, 74698); + } + + #[test] + fn example_part2() { + let input = include_str!("../example.txt"); + let result = part2(input); + assert_eq!(result, 71503); + } + + #[test] + fn input_part2() { + let input = include_str!("../input.txt"); + let result = part2(input); + assert_eq!(result, 27563421); + } +} |