This commit is contained in:
Daniel Flanagan 2022-12-07 08:52:05 -06:00
parent c53f8823dc
commit c461b0a958
Signed by untrusted user: lytedev-divvy
GPG key ID: 6D69CEEE4ABBCD82
2 changed files with 149 additions and 0 deletions

View file

@ -38,4 +38,8 @@ path = "src/day5.rs"
name = "day6"
path = "src/day6.rs"
[[bin]]
name = "day7"
path = "src/day7.rs"
[dependencies]

145
2022/rust/src/day7.rs Normal file
View file

@ -0,0 +1,145 @@
mod common;
use std::cell::RefCell;
use std::collections::HashMap;
use std::rc::Rc;
#[derive(Debug, Clone)]
struct DirTree {
files: HashMap<String, usize>,
subdirectories: HashMap<String, Rc<RefCell<Self>>>,
parent: Option<Rc<RefCell<Self>>>,
}
impl DirTree {
pub fn new(parent: Option<Rc<RefCell<Self>>>) -> Self {
Self {
files: HashMap::new(),
subdirectories: HashMap::new(),
parent,
}
}
pub fn root() -> Self {
Self::new(None)
}
pub fn size(&self) -> usize {
self.files.values().sum::<usize>()
+ self
.subdirectories
.values()
.map(|v| v.borrow().size())
.sum::<usize>()
}
}
struct DirTreeIterator {
dir_tree: Rc<RefCell<DirTree>>,
}
impl IntoIterator for DirTree {
type Item = Rc<RefCell<DirTree>>;
type IntoIter = std::iter::Chain<
std::iter::Once<Rc<RefCell<DirTree>>>,
std::collections::hash_map::Values<String, Rc<RefCell<DirTree>>>,
>;
fn into_iter(&self) -> Self::IntoIter {
std::iter::once(Rc::new(RefCell::new(*self)))
.into_iter()
.chain(self.subdirectories.values().into_iter())
}
}
type Input = Rc<RefCell<DirTree>>;
type Result = usize;
fn processed_input(input: &str) -> Input {
let root = Rc::new(RefCell::new(DirTree::root()));
let mut current_dir = Rc::clone(&root);
for l in input.lines() {
println!("{}", l);
if l == "$ cd /" {
current_dir = Rc::clone(&root);
} else if l == "$ ls" {
continue;
} else if l.starts_with("dir ") {
current_dir.borrow_mut().subdirectories.insert(
l[4..].to_string(),
Rc::new(RefCell::new(DirTree::new(Some(Rc::clone(&current_dir))))),
);
} else if l.starts_with("$ cd ..") {
let cur = Rc::clone(&current_dir);
current_dir = Rc::clone(&cur.borrow().parent.as_ref().unwrap());
} else if l.starts_with("$ cd ") {
let cur = Rc::clone(&current_dir);
current_dir = Rc::clone(&cur.borrow().subdirectories.get(&l[5..]).unwrap());
} else {
let mut args = l.split(' ');
let size = args.next().unwrap().parse::<usize>().unwrap();
current_dir
.borrow_mut()
.files
.insert(args.next().unwrap().to_string(), size);
}
}
return root;
}
fn part1(input: &Input) -> Result {
100
}
fn part2(input: &Input) -> Result {
0
}
fn main() {
let input_text = common::day_input(7);
eprintln!("{}\n\nAbove is your input file.\n\n", input_text);
let input = processed_input(&input_text);
common::show_answers(&part1(&input), &part2(&input))
// common::show_both_answers(&both_parts(&input))
}
// fn both_parts(input: &Input) -> (Result, Result) {
// (0, 0)
// }
#[cfg(test)]
mod tests {
use super::*;
const TEST_INPUT: &str = "$ cd /
$ ls
dir a
14848514 b.txt
8504156 c.dat
dir d
$ cd a
$ ls
dir e
29116 f
2557 g
62596 h.lst
$ cd e
$ ls
584 i
$ cd ..
$ cd ..
$ cd d
$ ls
4060174 j
8033020 d.log
5626152 d.ext
7214296 k";
#[test]
fn test() {
let input = processed_input(TEST_INPUT);
assert_eq!(part1(&input), 95437);
assert_eq!(part2(&input), 0);
// assert_eq!(both_parts(&input), (0, 0));
}
}