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

day14/1: optimized

Browse Source
This commit is contained in:
Ishan Jain 2023-12-14 18:22:08 +05:30
parent 7460612996
commit 3fa44cb9e2
Signed by: ishan
GPG Key ID: 0506DB2A1CC75C27
1 changed files with 18 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
src/day14/1.rs
View File

@ -5,16 +5,14 @@ extern crate test;
const INPUTS: [&str; 2] = [include_str!("./sample.txt"), include_str!("./input.txt")];
fn process(data: &str) -> usize {
let size = data.lines().count();
let mut data: Vec<u8> = data.bytes().collect();
let mut answer = 0;
let mut grid: Vec<Vec<u8>> = data.lines().map(|x| x.bytes().collect()).collect();
let m = grid.len();
let n = grid[0].len();
for i in 0..m {
for j in 0..n {
let c = grid[i][j];
for i in 0..size {
for j in 0..size {
let c = unsafe { *data.get_unchecked(size * i + j + i) };
if c != b'O' {
continue;
@ -22,21 +20,24 @@ fn process(data: &str) -> usize {
// move the rock north
let mut start = i;
while start > 0 && grid[start - 1][j] == b'.' {
while start > 0
&& unsafe { *data.get_unchecked(size * (start - 1) + j + start - 1) } == b'.'
{
start -= 1;
}
grid[i][j] = b'.';
grid[start][j] = b'O';
unsafe {
*data.get_unchecked_mut(size * i + j + i) = b'.';
*data.get_unchecked_mut(size * start + j + start) = b'O';
}
}
}
for (i, line) in grid.iter().enumerate() {
for &c in line {
if c == b'O' {
answer += m - i;
}
}
for i in 0..size {
let line = unsafe { data.get_unchecked(size * i + i..size * i + i + size) };
let count: usize = line.iter().filter(|&&x| x == b'O').count();
answer += count * (size - i);
}
answer