75 lines
2.2 KiB
Rust
75 lines
2.2 KiB
Rust
#![allow(dead_code)]
|
|
use std::collections::HashMap;
|
|
use std::fs::File;
|
|
use std::io;
|
|
use std::io::BufRead;
|
|
use std::path::Path;
|
|
use std::time::Instant;
|
|
|
|
#[derive(Debug, Clone)]
|
|
struct Folder {
|
|
folders: HashMap<String, u32>,
|
|
size: u32,
|
|
}
|
|
|
|
fn main() {
|
|
let now = Instant::now();
|
|
part_one();
|
|
let part_one_duration = now.elapsed();
|
|
part_two();
|
|
let part_two_duration = now.elapsed();
|
|
println!(
|
|
"p1: {}ms, p2: {}ms",
|
|
part_one_duration.as_millis(),
|
|
(part_two_duration - part_one_duration).as_millis()
|
|
);
|
|
}
|
|
|
|
fn read_lines<P>(filename: P) -> io::Result<io::Lines<io::BufReader<File>>>
|
|
where
|
|
P: AsRef<Path>,
|
|
{
|
|
// note that this discards a final newline
|
|
let file = File::open(filename)?;
|
|
Ok(io::BufReader::new(file).lines())
|
|
}
|
|
|
|
fn part_one() {
|
|
if let Ok(lines) = read_lines("./inputs/input") {
|
|
let mut sum = 0;
|
|
let mut stack: Vec<u32> = vec![0];
|
|
for line in lines {
|
|
if let Ok(command) = line {
|
|
match command.as_str() {
|
|
"$ cd /" => continue, // skip start
|
|
"$ ls" => continue, // we can assume this if we're not moving between directories
|
|
s if s.starts_with("dir ") => continue, // we'll get there when we `cd`
|
|
"$ cd .." => {
|
|
let f = stack.pop().unwrap();
|
|
if f < 100000 {
|
|
sum += f
|
|
}
|
|
let mut p = stack.pop().unwrap();
|
|
p += f;
|
|
stack.push(p);
|
|
}
|
|
s if s.starts_with("$ cd ") => {
|
|
stack.push(0);
|
|
}
|
|
s => {
|
|
// otherwise it must be a file
|
|
let mut f = stack.pop().unwrap();
|
|
let mut sp = s.split(" ");
|
|
let size = sp.next().unwrap().parse::<u32>().unwrap();
|
|
f += size;
|
|
stack.push(f);
|
|
}
|
|
};
|
|
}
|
|
}
|
|
println!("{:?}", sum);
|
|
}
|
|
}
|
|
|
|
fn part_two() {}
|