SebastianStork/advent-of-code
1
0
Fork 0
mirror of https://github.com/SebastianStork/advent-of-code.git synced 2026-01-21 12:11: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() {
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>> {
@ -14,24 +15,31 @@ fn parse_input(contents: &str) -> Vec<Vec<u8>> {
.collect()
}
fn total_joltage(banks: Vec<Vec<u8>>) -> u32 {
banks.iter().map(|bank| largest_joltage(bank)).sum()
fn total_joltage(banks: &[Vec<u8>], number_of_activate_batteries: usize) -> u64 {
banks
.iter()
.map(|bank| largest_joltage(bank, number_of_activate_batteries))
.sum()
}
fn largest_joltage(batteries: &[u8]) -> u32 {
let mut first = batteries[0];
let mut second = batteries[1];
fn largest_joltage(batteries: &[u8], number_of_activate_batteries: usize) -> u64 {
let mut active_batteries: Vec<u8> = batteries[0..number_of_activate_batteries].into();
for i in 1..batteries.len() - 1 {
if batteries[i] > first {
first = batteries[i];
second = batteries[i + 1];
} else if batteries[i + 1] > second {
second = batteries[i + 1];
for i in 1..batteries.len() - (number_of_activate_batteries - 1) {
for j in 0..number_of_activate_batteries {
if batteries[j + i] > active_batteries[j] {
active_batteries[j..number_of_activate_batteries]
.copy_from_slice(&batteries[j + i..number_of_activate_batteries + i]);
}
}
}
(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)]
@ -41,11 +49,15 @@ mod tests {
const TEST_INPUT: &str = "987654321111111
811111111111119
234234234234278
818181911112111
";
818181911112111";
#[test]
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);
}
}