aoc_4 fast version

benches/bench_aoc_4.rs

@@ -0,0 +1,29 @@
+use std::hint::black_box;
+use criterion::{criterion_group, criterion_main, Criterion};
+use aoc2025::aoc::aoc_4;
+
+fn bench_aoc_4_part1(c: &mut Criterion) {
+    let input = black_box(include_bytes!("../src/aoc/input/full_input_aoc4.txt"));
+    c.bench_function("bench_aoc4 part 1", |b| b.iter(|| aoc_4::solve_p1(input)));
+}
+fn bench_aoc_4_part1_f(c: &mut Criterion) {
+    let input = black_box(include_bytes!("../src/aoc/input/full_input_aoc4.txt"));
+    c.bench_function("bench_aoc4 part 1 fast", |b| b.iter(|| aoc_4::solve_p1_f(input)));
+}
+fn bench_aoc_4_part2(c: &mut Criterion) {
+    let input = black_box(include_str!("../src/aoc/input/full_input_aoc4.txt"));
+    c.bench_function("bench_aoc4 part 2", |b| b.iter(|| aoc_4::solve_p2(input)));
+}
+fn bench_aoc_4_part2_f(c: &mut Criterion) {
+    let input = black_box(include_bytes!("../src/aoc/input/full_input_aoc4.txt"));
+    c.bench_function("bench_aoc4 part 2", |b| b.iter(|| aoc_4::solve_p2_f(input)));
+}
+
+criterion_group!(
+    benches_p4,
+    bench_aoc_4_part1,
+    bench_aoc_4_part1_f,
+    bench_aoc_4_part2,
+    bench_aoc_4_part2_f
+);
+criterion_main!(benches_p4);

src/aoc/aoc_2.rs

@@ -1,28 +1,238 @@
-use std::io::BufRead;
-
-pub fn solve_p1(input: &str ) -> u64 {
+use strength_reduce::StrengthReducedU64;
+pub fn solve_p1(input: &str) -> u64 {
     let mut answer = 0;
     for line in input.split(',') {
-        let (num1, num2) = line.split_once('-').map(|n| {(n.0.parse::<u64>().unwrap(), n.1.parse::<u64>().unwrap())}).unwrap();
+        let (num1, num2) = line
+            .split_once('-')
+            .map(|n| (n.0.parse::<u64>().unwrap(), n.1.parse::<u64>().unwrap()))
+            .unwrap();
         for i in num1..=num2 {
             let text = i.to_string();
-            if text.len()%2 == 0 {
-                let splits = text.split_at(text.len()/2);
-                if splits.0==splits.1 {
-                    answer +=i;
+            if text.len() % 2 == 0 {
+                let splits = text.split_at(text.len() / 2);
+                if splits.0 == splits.1 {
+                    answer += i;
                 }
             }
         }
     }
 
-
     answer
 }
 
-pub fn solve_p2(input: &str ) -> u64 {
+const POW10: [u64; 10] = [
+    1, 10, 100, 1000, 10000, 100000, 1000000, 10000000, 100000000, 1000000000,
+];
+
+pub fn solve_p1_f(input: &str) -> u64 {
+    let mut total_answer = 0;
+
+    for line in input.split(',') {
+        let Some((s1, s2)) = line.split_once('-') else {
+            continue;
+        };
+        let n1: u64 = s1.parse().unwrap();
+        let n2: u64 = s2.parse().unwrap();
+
+        // Iterate through possible "half lengths".
+        // u64 max is ~1.8e19, so max length is 19.
+        // We only care about even lengths: 2, 4, 6... 18.
+        // So 'half_len' goes from 1 to 9.
+        let half_len = n1.ilog10() / 2 + 1;
+        let power_of_10 = POW10[half_len as usize];
+        let multiplier = power_of_10 + 1;
+
+        // X must be a number with exactly 'half_len' digits.
+        // Range for X: [10^(d-1), 10^d - 1]
+        let x_min_digits = POW10[(half_len - 1) as usize];
+        let x_max_digits = power_of_10 - 1;
+
+        // X must also satisfy: n1 <= X * multiplier <= n2
+        // Therefore: n1/multiplier <= X <= n2/multiplier
+        // We use ceiling for lower bound and floor for upper bound.
+        let x_min_val = (n1 + multiplier - 1) / multiplier;
+        let x_max_val = n2 / multiplier;
+
+        // Find the intersection of valid digits and valid values
+        let start = x_min_digits.max(x_min_val);
+        let end = x_max_digits.min(x_max_val);
+
+        if start <= end {
+            // Sum of numbers N = X * multiplier
+            // Sum = multiplier * Sum(X for X in start..=end)
+            // Sum(start..=end) = (start + end) * count / 2
+            let count = end - start + 1;
+            let sum_x = (start + end) * count / 2;
+            total_answer += sum_x * multiplier;
+        }
+    }
+
+    total_answer
+}
+
+pub fn solve_p1_bytes(bytes: &[u8]) -> u64 {
+    let mut total_answer = 0;
+    let mut i = 0;
+    let len = bytes.len();
+
+    while i < len {
+        // 1. Parse N1 and count its digits (c1)
+        let mut n1: u64 = 0;
+        let mut c1: usize = 0;
+        while i < len {
+            let b = bytes[i];
+            if b == b'-' {
+                i += 1;
+                break;
+            }
+            n1 = n1 * 10 + (b & 0x0F) as u64;
+            c1 += 1;
+            i += 1;
+        }
+
+        // 2. Parse N2 and count its digits (c2)
+        let mut n2: u64 = 0;
+        while i < len {
+            let b = bytes[i];
+            if b == b',' {
+                i += 1;
+                break;
+            }
+            n2 = n2 * 10 + (b & 0x0F) as u64;
+            i += 1;
+        }
+
+        let half_len = (c1 + 1) >> 1; // Bitwise division by 2
+
+        // --- INLINED MATH (No function call overhead) ---
+        let power_of_10 = POW10[half_len];
+        let multiplier = power_of_10 + 1;
+
+        let x_min_digits = POW10[half_len - 1];
+        let x_max_digits = power_of_10 - 1;
+
+        // Fast ceiling division: (n1 + m - 1) / m
+        let x_min_val = (n1 + multiplier - 1) / multiplier;
+        let x_max_val = n2 / multiplier;
+
+        // Branchless max/min
+        let start = if x_min_digits > x_min_val {
+            x_min_digits
+        } else {
+            x_min_val
+        };
+        let end = if x_max_digits < x_max_val {
+            x_max_digits
+        } else {
+            x_max_val
+        };
+
+        if start <= end {
+            let count = end - start + 1;
+            // Formula: Sum = multiplier * (start + end) * count / 2
+            total_answer += multiplier * (start + end) * count / 2;
+        }
+    }
+
+    total_answer
+}
+
+pub fn solve_p1_unsafe(input: &str) -> u64 {
+    let mut total_answer = 0;
+
+    // Get raw pointers
+    let mut ptr = input.as_ptr();
+    // Calculate the end address
+    let end = unsafe { ptr.add(input.len()) };
+
+    while ptr < end {
+        // 1. Parse N1 (Unsafe, no bounds check)
+        let mut n1: u64 = 0;
+        let start_ptr = ptr;
+        loop {
+            // Read byte directly from memory
+            let b = unsafe { *ptr };
+            if b == b'-' {
+                ptr = unsafe { ptr.add(1) }; // Skip '-'
+                break;
+            }
+            // Fast ASCII conversion
+            n1 = n1 * 10 + (b & 0x0F) as u64; // b & 0x0F is faster than b - b'0'
+            ptr = unsafe { ptr.add(1) };
+        }
+
+        // Calculate digits of N1 using pointer difference (Instant)
+        // (Current ptr is at dash + 1) - (Start ptr) - 1 (for the dash)
+        let c1 = (ptr as usize - start_ptr as usize - 1) as u8;
+
+        // 2. Parse N2
+        let mut n2: u64 = 0;
+        // let start_ptr2 = ptr; // Not strictly needed unless you want c2
+        loop {
+            // Check if we hit end of string (EOF protection)
+            if ptr == end {
+                break;
+            }
+
+            let b = unsafe { *ptr };
+            if b == b',' {
+                ptr = unsafe { ptr.add(1) }; // Skip ','
+                break;
+            }
+
+            // Only valid digits here
+            n2 = n2 * 10 + (b & 0x0F) as u64;
+            ptr = unsafe { ptr.add(1) };
+        }
+
+        // --- LOGIC ---
+        // c1 is enough because we know the length difference is <= 1.
+        // If c1 is even, we check c1. If c1 is odd, we check c1 + 1 (which is c2).
+
+        // Trick: We need to check 'c1' if it's even, or 'c1+1' if c1 is odd.
+        // Let's just calculate the specific half_len we care about.
+
+        // If c1 is even (e.g. 2), we want half_len=1.
+        // If c1 is odd (e.g. 3), we want half_len=2 (because target must be len 4).
+        // Formula: half_len = (c1 + 1) / 2
+        let half_len = ((c1 + 1) >> 1) as usize;
+
+        let power_of_10 = unsafe { *POW10.get_unchecked(half_len) };
+        let multiplier = power_of_10 + 1;
+
+        let x_min_digits = unsafe { *POW10.get_unchecked(half_len - 1) };
+        let x_max_digits = power_of_10 - 1;
+
+        let x_min_val = (n1 + multiplier - 1).div_euclid(multiplier);
+        let x_max_val = n2 / multiplier;
+
+        let start = if x_min_digits > x_min_val {
+            x_min_digits
+        } else {
+            x_min_val
+        };
+        let end = if x_max_digits < x_max_val {
+            x_max_digits
+        } else {
+            x_max_val
+        };
+
+        if start <= end {
+            let count = end - start + 1;
+            total_answer += multiplier * (start + end) * count / 2;
+        }
+    }
+
+    total_answer
+}
+
+pub fn solve_p2(input: &str) -> u64 {
     let mut answer = 0;
     for line in input.split(',') {
-        let num_range = line.split('-').map(|num| { num.parse::<u64>().unwrap_or(0) }).collect::<Vec<u64>>();
+        let num_range = line
+            .split('-')
+            .map(|num| num.parse::<u64>().unwrap_or(0))
+            .collect::<Vec<u64>>();
         for i in num_range[0]..=num_range[1] {
             let text = i.to_string();
             for j in 1..=text.len() / 2 {

src/aoc/aoc_4.rs

@@ -0,0 +1,282 @@
+use std::simd::prelude::*;
+pub fn solve_p1(input: &[u8]) -> u64 {
+    let mut answer = 0;
+    let mut data:Vec<&u8> = Vec::with_capacity(32768);
+
+    let mut len = 141;
+
+    for line in input.split(|&x| {x == b'\n'}) {
+        let mut data_line: Vec<&u8> = line.iter().collect();
+        data_line.insert(0, &b'0');
+        data_line.insert(data_line.len(), &b'0');
+        len = data_line.len();
+        data.append(&mut data_line);
+    }
+    let size = 139;
+
+    for (i,char) in data.clone().into_iter().enumerate() {
+        let mut count = 0;
+        if *char == b'0' || *char == b'.' {
+            continue;
+        }
+        if char == &b'@' {
+            if i as i32 - len as i32 > 0 {
+                if data[i - (len-1)] == &b'@' {
+                    count += 1;
+                }
+                if data[i - len] == &b'@' {
+                    count += 1;
+                }
+                if data[i - (len + 1)] == &b'@' {
+                    count += 1;
+                }
+            }
+            if i + len < (len * size)  {
+                if data[i + len +1] == &b'@' {
+                    count += 1;
+                }
+                if data[i + len] == &b'@' {
+                    count += 1;
+                }
+                if data[i + len -1] == &b'@' {
+                    count += 1;
+                }
+            }
+            if data[i - 1] == &b'@' {
+                count += 1;
+            }
+            if data[i + 1] == &b'@' {
+                count += 1;
+            }
+            if count < 4 {
+                //println!("count = {}", count);
+                answer += 1;
+            }
+
+
+        }
+        count = 0;
+    }
+    answer
+}
+
+pub fn solve_p1_f(input: &[u8]) -> u64 {
+    let mut answer = 0;
+
+    let len = 140;
+    let target = SimdVec::splat(b'@');
+    const LANES: usize = 8;
+    type SimdVec = Simd<u8, LANES>;
+
+    for i in len+1..input.len()-len-1 {
+        if input[i] == b'@' {
+            let data= [input[i-1], input[i+1], input[i+len], input[i +len+1], input[i+len-1], input[i-(len+1)], input[i-(len -1)], input[i-len]];
+            let (pre, chk, post) = data.as_simd();
+            let count = chk[0].simd_eq(target).to_bitmask().count_ones();
+            if count <4 {
+                answer += 1;
+            }
+        }
+    }
+    for i in 1..len {
+        if input[i] == b'@' {
+            let data= [input[i-1], input[i+1], input[i+len], input[i +len +1], input[i+len -1], 0,0,0];
+            let (pre, chk, post) = data.as_simd();
+            let count = chk[0].simd_eq(target).to_bitmask().count_ones();
+            if count <4 {
+                answer += 1;
+            }
+        }
+    }
+    for i in input.len()-len..input.len()-1 {
+        if input[i] == b'@' {
+            let data= [input[i-1], input[i+1], input[i-len], input[i -(len +1)], input[i-(len-1)], 0,0,0];
+            let (pre, chk, post) = data.as_simd();
+            let count = chk[0].simd_eq(target).to_bitmask().count_ones();
+            if count <4 {
+
+                answer += 1;
+            }
+        }
+    }
+    if input[len] == b'@' {
+        let data= [input[len-1], input[len+1], input[0], 0, input[1], 0,0,0];
+        let (pre, chk, post) = data.as_simd();
+        let count = chk[0].simd_eq(target).to_bitmask().count_ones();
+        if count <4 {
+
+            answer += 1;
+        }
+    }
+    let idx = input.len()-len -1;
+    if input[idx] == b'@' {
+        let data= [input[idx-1], 0, input[idx - len], input[idx + len], input[idx - (len + 1)], input[idx + (len -1)],0,0];
+        let (pre, chk, post) = data.as_simd();
+        let count = chk[0].simd_eq(target).to_bitmask().count_ones();
+        if count <4 {
+
+            answer += 1;
+        }
+    }
+    if input[0] == b'@'{
+        answer += 1;
+    }
+    if input[input.len()-1] == b'@'{
+        answer += 1;
+    }
+    answer
+}
+
+pub fn solve_p2(input: &str) -> u16 {
+    let mut answer = 0;
+    let mut data:Vec<Vec<char>> = Vec::with_capacity(256);
+
+    for line in input.split('\n') {
+        let mut data_line: Vec<char> = line.chars().collect();
+        data_line.insert(0, '0');
+        data_line.insert(data_line.len(), '0');
+        data.push(data_line);
+    }
+    let len = data[0].len();
+    let size = data.len();
+
+    let mut data: Vec<&char> = data.iter().flatten().collect();
+    let mut ans = 0;
+    loop {
+        for (i, char) in data.clone().iter().enumerate() {
+            let mut count = 0;
+
+            if *char == &'0' || *char == &'.' {
+                continue;
+            }
+            if *char == &'@' {
+                if i as i32 - len as i32 > 0 {
+                    if data[i - (len - 1)] == &'@' {
+                        count += 1;
+                    }
+                    if data[i - len] == &'@' {
+                        count += 1;
+                    }
+                    if data[i - (len + 1)] == &'@' {
+                        count += 1;
+                    }
+                }
+                if i + len < (len * size) {
+                    if data[i + len + 1] == &'@' {
+                        count += 1;
+                    }
+                    if data[i + len] == &'@' {
+                        count += 1;
+                    }
+                    if data[i + len - 1] == &'@' {
+                        count += 1;
+                    }
+                }
+                if data[i - 1] == &'@' {
+                    count += 1;
+                }
+                if data[i + 1] == &'@' {
+                    count += 1;
+                }
+                if count < 4 {
+                    data[i] = &'-';
+                    ans += 1;
+                }
+
+
+            }
+            count = 0;
+        }
+        if ans == 0 {
+            break;
+        } else {
+            answer += ans;
+        }
+        ans = 0;
+    }
+    answer
+}
+
+pub fn solve_p2_f(input: &[u8]) -> u64 {
+    let mut input =  input.to_vec();
+    let mut answer = 0;
+
+    let len = 140;
+    let target = SimdVec::splat(b'@');
+    const LANES: usize = 8;
+    type SimdVec = Simd<u8, LANES>;
+    let mut ans = 0;
+    loop {
+        for i in len + 1..input.len() - len - 1 {
+            if input[i] == b'@' {
+                let data = [input[i - 1], input[i + 1], input[i + len], input[i + len + 1], input[i + len - 1], input[i - (len + 1)], input[i - (len - 1)], input[i - len]];
+                let (pre, chk, post) = data.as_simd();
+                let count = chk[0].simd_eq(target).to_bitmask().count_ones();
+                if count < 4 {
+                    ans += 1;
+                    input[i] = b'.';
+                }
+            }
+        }
+        for i in 1..len {
+            if input[i] == b'@' {
+                let data = [input[i - 1], input[i + 1], input[i + len], input[i + len + 1], input[i + len - 1], 0, 0, 0];
+                let (pre, chk, post) = data.as_simd();
+                let count = chk[0].simd_eq(target).to_bitmask().count_ones();
+                if count < 4 {
+                    ans += 1;
+                    input[i] = b'.';
+                }
+            }
+        }
+        for i in input.len() - len..input.len() - 1 {
+            if input[i] == b'@' {
+                let data = [input[i - 1], input[i + 1], input[i - len], input[i - (len + 1)], input[i - (len - 1)], 0, 0, 0];
+                let (pre, chk, post) = data.as_simd();
+                let count = chk[0].simd_eq(target).to_bitmask().count_ones();
+                if count < 4 {
+                    ans += 1;
+                    input[i] = b'.';
+                }
+            }
+        }
+        if input[len] == b'@' {
+            let data = [input[len - 1], input[len + 1], input[0], 0, input[1], 0, 0, 0];
+            let (pre, chk, post) = data.as_simd();
+            let count = chk[0].simd_eq(target).to_bitmask().count_ones();
+            if count < 4 {
+                ans += 1;
+                input[len] = b'.';
+            }
+        }
+        let idx = input.len() - len - 1;
+        if input[idx] == b'@' {
+            let data = [input[idx - 1], 0, input[idx - len], input[idx + len], input[idx - (len + 1)], input[idx + (len - 1)], 0, 0];
+            let (pre, chk, post) = data.as_simd();
+            let count = chk[0].simd_eq(target).to_bitmask().count_ones();
+            if count <4 {
+
+                ans += 1;
+                input[idx] = b'.';
+            }
+        }
+        if input[0] == b'@'{
+            ans += 1;
+            input[0] = b'.';
+        }
+        let idx = input.len()-1;
+        if input[idx] == b'@'{
+            ans += 1;
+            input[idx] = b'.';
+        }
+        if ans == 0 {
+            break;
+        }else {
+            answer += ans;
+            ans = 0;
+        }
+    }
+
+
+    answer
+}

src/aoc/input/test_input_aoc4.txt

@@ -0,0 +1,10 @@
+..@@.@@@@.
+@@@.@.@.@@
+@@@@@.@.@@
+@.@@@@..@.
+@@.@@@@.@@
+.@@@@@@@.@
+.@.@.@.@@@
+@.@@@.@@@@
+.@@@@@@@@.
+@.@.@@@.@.

src/main.rs

@@ -1,13 +1,24 @@
+#![feature(portable_simd)] // Enable the nightly feature
 pub mod aoc;
 use crate::aoc::*;
 
 fn main() {
+    /*
     let input = include_bytes!("aoc/input/full_input_aoc3.txt");
     let answer = aoc_3::solve_p1(input);
     println!("{}", answer);
     let input2 = include_bytes!("aoc/input/full_input_aoc3.txt");
     let answer = aoc_3::solve_p1_fast(input2);
     println!("{}", answer);
+     */
+
+    let input = include_bytes!("aoc/input/full_input_aoc4.txt");
+    let answer = aoc_4::solve_p2_f(input);
+    println!("{}", answer);
+
+    let input = include_str!("aoc/input/full_input_aoc4.txt");
+    let answer = aoc_4::solve_p2(input);
+    println!("{}", answer);
 }
 
 #[test]