SebastianStork/advent-of-code
1
0
Fork 0
mirror of https://github.com/SebastianStork/advent-of-code.git synced 2026-01-21 13:21:34 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Solve 2025 day 3 part 2 in rust

Browse source
This commit is contained in:
SebastianStork 2025-12-03 19:09:15 +01:00
parent 586e674325
commit c7683b3600
Signed by: SebastianStork
SSH key fingerprint: SHA256:iEM011ogNMG1q8+U500adGu/9rpPuZ2KnFtbdLeqTiI
1 changed files with 28 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

44
2025/rust/day-03/src/main.rs
View file

@ -3,7 +3,8 @@ use std::fs;
fn main() { fn main() {
let banks = parse_input(&fs::read_to_string("../../inputs/03.txt").unwrap()); let banks = parse_input(&fs::read_to_string("../../inputs/03.txt").unwrap());
println!("Total output joltage: {}", total_joltage(banks)); println!("Total output joltage: {}", total_joltage(&banks, 2));
println!("New total output joltage: {}", total_joltage(&banks, 12));
} }
fn parse_input(contents: &str) -> Vec<Vec<u8>> { fn parse_input(contents: &str) -> Vec<Vec<u8>> {
@ -14,24 +15,31 @@ fn parse_input(contents: &str) -> Vec<Vec<u8>> {
.collect() .collect()
} }
fn total_joltage(banks: Vec<Vec<u8>>) -> u32 { fn total_joltage(banks: &[Vec<u8>], number_of_activate_batteries: usize) -> u64 {
banks.iter().map(|bank| largest_joltage(bank)).sum() banks
.iter()
.map(|bank| largest_joltage(bank, number_of_activate_batteries))
.sum()
} }
fn largest_joltage(batteries: &[u8]) -> u32 { fn largest_joltage(batteries: &[u8], number_of_activate_batteries: usize) -> u64 {
let mut first = batteries[0]; let mut active_batteries: Vec<u8> = batteries[0..number_of_activate_batteries].into();
let mut second = batteries[1];
for i in 1..batteries.len() - 1 { for i in 1..batteries.len() - (number_of_activate_batteries - 1) {
if batteries[i] > first { for j in 0..number_of_activate_batteries {
first = batteries[i]; if batteries[j + i] > active_batteries[j] {
second = batteries[i + 1]; active_batteries[j..number_of_activate_batteries]
} else if batteries[i + 1] > second { .copy_from_slice(&batteries[j + i..number_of_activate_batteries + i]);
second = batteries[i + 1]; }
} }
} }
(first * 10 + second) as u32 active_batteries
.iter()
.rev()
.enumerate()
.map(|(i, joltage)| *joltage as u64 * 10_u64.pow(i as u32))
.sum()
} }
#[cfg(test)] #[cfg(test)]
@ -41,11 +49,15 @@ mod tests {
const TEST_INPUT: &str = "987654321111111 const TEST_INPUT: &str = "987654321111111
811111111111119 811111111111119
234234234234278 234234234234278
818181911112111 818181911112111";
";
#[test] #[test]
fn test_part1() { fn test_part1() {
assert_eq!(total_joltage(parse_input(TEST_INPUT)), 357); assert_eq!(total_joltage(&parse_input(TEST_INPUT), 2), 357);
}
#[test]
fn test_part2() {
assert_eq!(total_joltage(&parse_input(TEST_INPUT), 12), 3121910778619);
} }
} }