ishan/aoc2023
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

day13: optimized

Browse Source
This commit is contained in:
Ishan Jain 2023-12-13 21:01:36 +05:30
parent f667e05c91
commit 5707b7cacc
Signed by: ishan
GPG Key ID: 0506DB2A1CC75C27
2 changed files with 34 additions and 87 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
src/day13/1.rs
View File

@ -7,39 +7,25 @@ const INPUTS: [&str; 2] = [include_str!("./sample.txt"), include_str!("./input.t
fn process(data: &str) -> usize { fn process(data: &str) -> usize {
let mut answer = 0; let mut answer = 0;
let mut horizontal_values: Vec<u32> = Vec::with_capacity(30);
let mut vertical_values: Vec<u32> = Vec::with_capacity(30);
for line in data.split("\n\n") { for line in data.split("\n\n") {
let mut horiz_set = vec![];
for line in line.lines() { for line in line.lines() {
let line_mask = line.chars().rev().enumerate().fold(0, |mut a, (i, x)| { let line_mask = line.bytes().fold(0u32, |a, x| a << 1 | (b'#' == x) as u32);
if let '#' = x {
a |= 1 << (i)
}
a
});
horiz_set.push(line_mask); horizontal_values.push(line_mask);
} vertical_values.resize(line.len(), 0);
let mut vert_set = vec![]; for (c, v) in line.bytes().zip(vertical_values.iter_mut()) {
*v <<= 1;
for (i, line) in line.lines().rev().enumerate() { *v |= (c == b'#') as u32;
for (j, c) in line.chars().enumerate() {
if vert_set.len() <= j {
vert_set.push(0);
}
if c == '#' {
vert_set[j] |= 1 << i;
}
} }
} }
let horiz_max = (0..horiz_set.len()) let horiz_max = (0..horizontal_values.len())
.rev()
.map(|j| { .map(|j| {
let (a, b) = horiz_set.split_at(j); let (a, b) = horizontal_values.split_at(j);
if a.iter().rev().zip(b.iter()).any(|(a, b)| a != b) { if a.iter().rev().zip(b.iter()).any(|(a, b)| a != b) {
return 0; return 0;
@ -50,10 +36,9 @@ fn process(data: &str) -> usize {
.max() .max()
.unwrap_or_default(); .unwrap_or_default();
let vertical_max = (0..vert_set.len()) let vertical_max = (0..vertical_values.len())
.rev()
.map(|i| { .map(|i| {
let (a, b) = vert_set.split_at(i); let (a, b) = vertical_values.split_at(i);
if a.iter().rev().zip(b.iter()).any(|(a, b)| a != b) { if a.iter().rev().zip(b.iter()).any(|(a, b)| a != b) {
return 0; return 0;
@ -63,9 +48,11 @@ fn process(data: &str) -> usize {
}) })
.max() .max()
.unwrap_or_default(); .unwrap_or_default();
answer += 100 * horiz_max;
answer += vertical_max; answer += vertical_max;
answer += 100 * horiz_max;
horizontal_values.clear();
vertical_values.clear();
} }
answer answer

76
src/day13/2.rs
View File

@ -7,49 +7,29 @@ const INPUTS: [&str; 2] = [include_str!("./sample.txt"), include_str!("./input.t
fn process(data: &str) -> usize { fn process(data: &str) -> usize {
let mut answer = 0; let mut answer = 0;
let mut horizontal_values: Vec<u32> = Vec::with_capacity(50);
let mut vertical_values: Vec<u32> = Vec::with_capacity(50);
for line in data.split("\n\n") { for line in data.split("\n\n") {
let mut horiz_set = vec![];
for line in line.lines() { for line in line.lines() {
let line_mask = line.chars().rev().enumerate().fold(0, |mut a, (i, x)| { let line_mask = line.bytes().fold(0u32, |a, x| a << 1 | (b'#' == x) as u32);
if let '#' = x {
a |= 1 << (i)
}
a
});
horiz_set.push(line_mask); horizontal_values.push(line_mask);
} vertical_values.resize(line.len(), 0);
let mut vert_set = vec![]; for (c, v) in line.bytes().zip(vertical_values.iter_mut()) {
*v <<= 1;
for (i, line) in line.lines().rev().enumerate() { *v |= (c == b'#') as u32;
for (j, c) in line.chars().enumerate() {
if vert_set.len() <= j {
vert_set.push(0);
}
if c == '#' {
vert_set[j] |= 1 << i;
}
} }
} }
let horiz_max = (0..horiz_set.len()) let horiz_max = (0..horizontal_values.len())
.rev()
.map(|j| { .map(|j| {
let (a, b) = horiz_set.split_at(j); let (a, b) = horizontal_values.split_at(j);
let mut error_count = 0; let mut error_count = 0;
for (a, b) in a.iter().rev().zip(b.iter()) {
for x in 0..31 {
let a = a & (1 << x);
let b = b & (1 << x);
if a != b { for (a, b) in a.iter().rev().zip(b.iter()) {
error_count += 1; error_count += (a ^ b).count_ones();
}
}
if error_count > 1 { if error_count > 1 {
return 0; return 0;
@ -65,20 +45,12 @@ fn process(data: &str) -> usize {
.max() .max()
.unwrap_or_default(); .unwrap_or_default();
let vertical_max = (0..vert_set.len()) let vertical_max = (0..vertical_values.len())
.rev()
.map(|i| { .map(|i| {
let (a, b) = vert_set.split_at(i); let (a, b) = vertical_values.split_at(i);
let mut error_count = 0; let mut error_count = 0;
for (a, b) in a.iter().rev().zip(b.iter()) { for (a, b) in a.iter().rev().zip(b.iter()) {
for x in 0..31 { error_count += (a ^ b).count_ones();
let a = a & (1 << x);
let b = b & (1 << x);
if a != b {
error_count += 1;
}
}
if error_count > 1 { if error_count > 1 {
return 0; return 0;
@ -96,26 +68,14 @@ fn process(data: &str) -> usize {
answer += 100 * horiz_max; answer += 100 * horiz_max;
answer += vertical_max; answer += vertical_max;
horizontal_values.clear();
vertical_values.clear();
} }
answer answer
} }
fn fmt(x: i32, length: usize) -> String {
let mut out = String::new();
for c in (0..length).rev() {
let c = x & (1 << c);
if c >= 1 {
out.push('#');
} else {
out.push('.');
}
}
out
}
fn main() { fn main() {
for input in INPUTS.iter() { for input in INPUTS.iter() {
println!("total = {}", process(input)); println!("total = {}", process(input));