day07 Add part 1
I don't know, it doesn't feel like this is proper code. Referencing the same object multiple times is just so much of a hassle and this code looks pretty unsafe with all the unwraps. In the end it turns out that I don't need a tree, but need to iterate over all directories instead, which is again something quite hard for me, because I don't know how to build iterators "manually".
This commit is contained in:
parent
01789441db
commit
2620d59a8f
|
@ -0,0 +1,7 @@
|
|||
# This file is automatically @generated by Cargo.
|
||||
# It is not intended for manual editing.
|
||||
version = 3
|
||||
|
||||
[[package]]
|
||||
name = "day07"
|
||||
version = "0.1.0"
|
|
@ -0,0 +1,8 @@
|
|||
[package]
|
||||
name = "day07"
|
||||
version = "0.1.0"
|
||||
edition = "2021"
|
||||
|
||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||
|
||||
[dependencies]
|
|
@ -0,0 +1,160 @@
|
|||
use std::cell::RefCell;
|
||||
use std::error;
|
||||
use std::fs::File;
|
||||
use std::io::BufRead;
|
||||
use std::io::BufReader;
|
||||
use std::rc::{Rc, Weak};
|
||||
|
||||
|
||||
enum FsEntry {
|
||||
Dir(Rc<FsDir>),
|
||||
File(FsFile)
|
||||
}
|
||||
|
||||
struct FsDir {
|
||||
name: String,
|
||||
children: RefCell<Vec<FsEntry>>,
|
||||
parent: Option<Weak<FsDir>>
|
||||
}
|
||||
|
||||
impl FsDir {
|
||||
fn _compute_size(&self) -> u32 {
|
||||
let mut size: u32 = 0;
|
||||
let children = self.children.borrow_mut();
|
||||
for child in children.iter() {
|
||||
match child {
|
||||
FsEntry::File(f) => size += f.size,
|
||||
FsEntry::Dir(d) => {
|
||||
size += d._compute_size();
|
||||
},
|
||||
};
|
||||
}
|
||||
size
|
||||
}
|
||||
|
||||
fn _show(&self, prefix: String) {
|
||||
println!("{}{}", &prefix, &self.name);
|
||||
let children = self.children.borrow_mut();
|
||||
for child in children.iter() {
|
||||
match child {
|
||||
FsEntry::File(f) => println!("{}{}", &prefix, &f.size),
|
||||
FsEntry::Dir(d) => d._show(format!("{} ", &prefix)),
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
fn get_dir_sizes(&self) -> Vec<u32> {
|
||||
let mut sizes = vec![];
|
||||
let mut size: u32 = 0;
|
||||
let children = self.children.borrow_mut();
|
||||
for child in children.iter() {
|
||||
match child {
|
||||
FsEntry::File(f) => size += f.size,
|
||||
FsEntry::Dir(d) => {
|
||||
let d_vec = d.get_dir_sizes();
|
||||
size += d_vec[d_vec.len() - 1];
|
||||
sizes.extend_from_slice(&d_vec);
|
||||
},
|
||||
};
|
||||
}
|
||||
sizes.push(size);
|
||||
sizes
|
||||
}
|
||||
}
|
||||
|
||||
struct FsFile {
|
||||
size: u32
|
||||
}
|
||||
|
||||
fn main() -> Result<(), Box<dyn error::Error>> {
|
||||
let f = File::open("input.txt")?;
|
||||
let reader = BufReader::new(f);
|
||||
|
||||
let fs_tree: Rc<FsDir> = Rc::new(FsDir { children: RefCell::new(vec![]), parent: None, name: String::from("/") });
|
||||
let mut current_dir = Rc::downgrade(&fs_tree);
|
||||
|
||||
for line in reader.lines() {
|
||||
let line = line?;
|
||||
if &line[0..1] == "$" {
|
||||
// parse the command
|
||||
let mut cmd_iter = line[2..].split_whitespace();
|
||||
match cmd_iter.next() {
|
||||
Some("cd") => {
|
||||
// println!("{}", &line);
|
||||
match cmd_iter.next() {
|
||||
Some("/") => current_dir = Rc::downgrade(&fs_tree),
|
||||
Some("..") => {
|
||||
let mut new_dir = None;
|
||||
let strong = current_dir.upgrade().unwrap();
|
||||
match &strong.parent {
|
||||
Some(x) => {
|
||||
let strong_x = x.upgrade().unwrap();
|
||||
let rc_c = Rc::clone(&strong_x);
|
||||
new_dir = Some(Rc::downgrade(&rc_c));
|
||||
},
|
||||
None => (),
|
||||
};
|
||||
match new_dir {
|
||||
Some(d) => current_dir = d,
|
||||
None => panic!("Could not find parent dir in '{}'", current_dir.upgrade().unwrap().name),
|
||||
};
|
||||
},
|
||||
Some(x) => {
|
||||
let mut new_dir = None;
|
||||
let strong = current_dir.upgrade().unwrap();
|
||||
let children = strong.children.borrow_mut();
|
||||
for child in children.iter() {
|
||||
if let FsEntry::Dir(d) = child {
|
||||
if d.name == x {
|
||||
new_dir = Some(Rc::clone(&d));
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
match new_dir {
|
||||
Some(d) => current_dir = Rc::downgrade(&d),
|
||||
None => panic!("Could not find dir '{x}' in '{}'", current_dir.upgrade().unwrap().name),
|
||||
}
|
||||
},
|
||||
None => panic!("No argument to 'cd': {}", line),
|
||||
};
|
||||
// println!("{}", current_dir.upgrade().unwrap().name);
|
||||
},
|
||||
Some("ls") => {
|
||||
// we can ignore this, as we read everything not being a command into the
|
||||
// current directory anyways
|
||||
}
|
||||
Some(x) => panic!("Unsupported command: {}", x),
|
||||
None => panic!("Could not split into a command: {}", line),
|
||||
};
|
||||
} else {
|
||||
// parse the list of fs entries
|
||||
let mut entry_iter = line.split_whitespace();
|
||||
match entry_iter.next() {
|
||||
Some("dir") => {
|
||||
let new_dir = Rc::new(FsDir {name: String::from(entry_iter.next().unwrap()),
|
||||
children: RefCell::new(vec![]), parent: Some(current_dir.clone()) });
|
||||
let strong = current_dir.upgrade().unwrap();
|
||||
let mut children = strong.children.borrow_mut();
|
||||
children.push(FsEntry::Dir(new_dir));
|
||||
},
|
||||
Some(size) => {
|
||||
let f = FsFile {size: size.parse().unwrap()};
|
||||
let strong = current_dir.upgrade().unwrap();
|
||||
let mut children = strong.children.borrow_mut();
|
||||
children.push(FsEntry::File(f));
|
||||
},
|
||||
_ => panic!("Unknown fs entry: {line}"),
|
||||
};
|
||||
}
|
||||
}
|
||||
drop(current_dir);
|
||||
// fs_tree.show(String::from(""));
|
||||
|
||||
// compute sizes
|
||||
let sizes = fs_tree.get_dir_sizes();
|
||||
println!("{}", sizes.iter().filter(|x| **x < 100000).sum::<u32>());
|
||||
|
||||
println!("Hello, world!");
|
||||
Ok(())
|
||||
}
|
Loading…
Reference in New Issue