From b3c0441d2168227ce50bca637aa332f1d06d6f52 Mon Sep 17 00:00:00 2001 From: June McEnroe Date: Sat, 10 Dec 2016 18:07:27 -0500 Subject: Refactor day 9 --- src/bin/day09.rs | 45 ++++++++++++++++----------------------------- 1 file changed, 16 insertions(+), 29 deletions(-) (limited to 'src') diff --git a/src/bin/day09.rs b/src/bin/day09.rs index fb3046f..0c6ca98 100644 --- a/src/bin/day09.rs +++ b/src/bin/day09.rs @@ -1,17 +1,19 @@ use std::io::{self, Read}; -use std::str::FromStr; -struct Chunk { - data: String, - repeat: u32, +enum Chunk<'a> { + Literal(&'a str), + Repeat(u32, &'a str), } -impl Chunk { +impl<'a> Chunk<'a> { fn len(&self) -> usize { - self.data.len() * self.repeat as usize + match *self { + Chunk::Literal(s) => s.len(), + Chunk::Repeat(n, s) => s.len() * n as usize, + } } - fn parse(s: &str) -> Result<(Self, &str), ()> { + fn parse(s: &'a str) -> Result<(Self, &'a str), ()> { if s.starts_with('(') { let mut iter = s[1..].splitn(3, |ch| ch == 'x' || ch == ')'); @@ -19,39 +21,24 @@ impl Chunk { let repeat = iter.next().ok_or(())?.parse().map_err(|_| ())?; let (data, rest) = iter.next().ok_or(())?.split_at(len); - let chunk = Chunk { - data: data.to_owned(), - repeat: repeat, - }; - - Ok((chunk, rest)) + Ok((Chunk::Repeat(repeat, data), rest)) } else { let paren = s.find('(').unwrap_or(s.len()); - - let chunk = Chunk { - data: s[..paren].to_owned(), - repeat: 1, - }; - let rest = &s[paren..]; - - Ok((chunk, rest)) + Ok((Chunk::Literal(&s[..paren]), &s[paren..])) } } } -struct File { - chunks: Vec, +struct File<'a> { + chunks: Vec>, } -impl File { +impl<'a> File<'a> { fn len(&self) -> usize { self.chunks.iter().map(Chunk::len).sum() } -} -impl FromStr for File { - type Err = (); - fn from_str(mut s: &str) -> Result { + fn parse(mut s: &'a str) -> Result { let mut chunks = Vec::new(); while !s.is_empty() { @@ -65,7 +52,7 @@ impl FromStr for File { } fn solve(input: &str) -> usize { - let file: File = input.parse().unwrap(); + let file = File::parse(input).unwrap(); file.len() } -- cgit 1.4.1