ishan/aoc2022
1
0
Fork 0
Code Releases Activity

Optimized day 7 part 1

Browse Source
This commit is contained in:
Ishan Jain 2022-12-07 17:16:39 +05:30
parent bd982a21c5
commit 26ae929eca
Signed by: ishan
GPG Key ID: 0506DB2A1CC75C27
1 changed files with 13 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/main.rs
View File

@ -1,5 +1,4 @@
#![feature(test)]
extern crate test;
const INPUTS: [&str; 2] = [
@ -23,21 +22,18 @@ fn main() {
}
fn solution(input: impl Iterator<Item = &'static str>) -> u32 {
let mut out = vec![];
let mut stack = vec![];
let mut out = Vec::with_capacity(100);
let mut stack = Vec::with_capacity(100);
let mut current_folder_size = 0;
for line in input {
if line.starts_with("$ ls") || line.starts_with("dir") {
continue;
}
match &line[..4] {
"$ ls" | "dir " => continue,
if line.starts_with("$ cd") {
let dir = line.trim_start_matches("$ cd ");
match dir {
".." => {
"$ cd" => match &line[5..6] {
// we are supposed to match on .. but this is fine
"." => {
let v = stack.pop().unwrap();
out.push(current_folder_size);
current_folder_size += v;
@ -49,14 +45,15 @@ fn solution(input: impl Iterator<Item = &'static str>) -> u32 {
stack.push(current_folder_size);
current_folder_size = 0;
}
}
} else {
},
_ => {
let (size, _) = line.split_once(' ').unwrap();
let size = size.parse::<u32>().unwrap();
current_folder_size += size;
}
}
}
while let Some(v) = stack.pop() {
out.push(current_folder_size);