aoc_4 fast version
This commit is contained in:
parent
f28c4bb28d
commit
2429d7a4dd
29
benches/bench_aoc_4.rs
Normal file
29
benches/bench_aoc_4.rs
Normal file
@ -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);
|
||||
282
src/aoc/aoc_4.rs
Normal file
282
src/aoc/aoc_4.rs
Normal file
@ -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
|
||||
}
|
||||
10
src/aoc/input/test_input_aoc4.txt
Normal file
10
src/aoc/input/test_input_aoc4.txt
Normal file
@ -0,0 +1,10 @@
|
||||
..@@.@@@@.
|
||||
@@@.@.@.@@
|
||||
@@@@@.@.@@
|
||||
@.@@@@..@.
|
||||
@@.@@@@.@@
|
||||
.@@@@@@@.@
|
||||
.@.@.@.@@@
|
||||
@.@@@.@@@@
|
||||
.@@@@@@@@.
|
||||
@.@.@@@.@.
|
||||
11
src/main.rs
11
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]
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user