ishan/aoc2022
1
0
Fork 0
Code Releases Activity

clean up day 14

Browse Source
This commit is contained in:
Ishan Jain 2022-12-14 15:52:56 +05:30
parent 02b57851a9
commit 7f255cab2c
Signed by: ishan
GPG Key ID: 0506DB2A1CC75C27
2 changed files with 13 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/day14/1.rs
View File

@ -78,13 +78,12 @@ fn solution((mut input, lowest): ([Node; ARR_SIZE], usize)) -> usize {
};
}
if py < lowest {
input[py * ROW_SIZE + px] = Node::Sand;
count += 1;
} else {
input[py * ROW_SIZE + px] = Node::Empty;
if py >= lowest {
break;
}
input[py * ROW_SIZE + px] = Node::Sand;
count += 1;
}
count

34
src/day14/2.rs
View File

@ -6,7 +6,7 @@ const INPUTS: [&str; 2] = [include_str!("./sample.txt"), include_str!("./input.t
const ROW_SIZE: usize = 700;
const ARR_SIZE: usize = 490000;
fn parse(input: &str) -> ([Node; ARR_SIZE], usize) {
fn parse(input: &str) -> [Node; ARR_SIZE] {
let mut grid = [Node::Empty; ARR_SIZE];
let mut lowest = 0;
@ -51,7 +51,7 @@ fn parse(input: &str) -> ([Node; ARR_SIZE], usize) {
grid[(lowest + 2) * ROW_SIZE + i] = Node::Rock;
}
(grid, lowest + 2)
grid
}
#[derive(Debug, Copy, Eq, PartialEq, Clone)]
@ -61,48 +61,32 @@ enum Node {
Sand,
}
fn solution((mut input, lowest): ([Node; ARR_SIZE], usize)) -> usize {
fn solution(mut input: [Node; ARR_SIZE]) -> usize {
let mut count = 1; // Include the source of sand in this count
loop {
let (mut px, mut py) = (500, 0);
let mut pmoved = false;
loop {
if py >= lowest {
input[py * ROW_SIZE + px] = Node::Empty;
break;
}
let moved = if input[(py + 1) * ROW_SIZE + px] == Node::Empty {
if input[(py + 1) * ROW_SIZE + px] == Node::Empty {
py += 1;
true
} else if input[(py + 1) * ROW_SIZE + px - 1] == Node::Empty {
py += 1;
px -= 1;
true
} else if input[(py + 1) * ROW_SIZE + px + 1] == Node::Empty {
py += 1;
px += 1;
true
} else {
false
};
if moved {
pmoved = true;
} else {
break;
}
};
}
if px == 500 && py == 0 {
break;
}
if pmoved {
input[py * ROW_SIZE + px] = Node::Sand;
count += 1;
} else {
input[py * ROW_SIZE + px] = Node::Empty;
break;
}
}
count