diff options
author | June McEnroe <june@causal.agency> | 2016-12-07 22:51:33 -0500 |
---|---|---|
committer | June McEnroe <june@causal.agency> | 2020-11-22 00:14:25 -0500 |
commit | 78896ceed1dcca5b8f7921bee964bb8ed0f4181a (patch) | |
tree | 82225fde1c088ba679694825cb2312378ff9466f /src/bin/day07.rs | |
parent | Refactor day 7 (diff) | |
download | aoc-78896ceed1dcca5b8f7921bee964bb8ed0f4181a.tar.gz aoc-78896ceed1dcca5b8f7921bee964bb8ed0f4181a.zip |
Day 7 part 2
Diffstat (limited to '')
-rw-r--r-- | src/bin/day07.rs | 57 |
1 files changed, 54 insertions, 3 deletions
diff --git a/src/bin/day07.rs b/src/bin/day07.rs index 305fd8f..b4967ff 100644 --- a/src/bin/day07.rs +++ b/src/bin/day07.rs @@ -11,6 +11,26 @@ fn has_abba(s: &str) -> bool { }) } +fn abas(s: &str) -> Vec<&[u8]> { + s.as_bytes() + .windows(3) + .filter(|window| { + window[0] == window[2] + && window[0] != window[1] + }) + .collect() +} + +fn has_bab(s: &str, aba: &[u8]) -> bool { + s.as_bytes() + .windows(3) + .any(|window| { + window[0] == aba[1] + && window[1] == aba[0] + && window[2] == aba[1] + }) +} + #[derive(Default)] struct Ip { supernet: Vec<String>, @@ -22,6 +42,17 @@ impl Ip { self.supernet.iter().any(|s| has_abba(s)) && !self.hypernet.iter().any(|s| has_abba(s)) } + + fn supports_ssl(&self) -> bool { + self.supernet + .iter() + .flat_map(|s| abas(s)) + .any(|aba| { + self.hypernet + .iter() + .any(|s| has_bab(s, aba)) + }) + } } impl FromStr for Ip { @@ -41,7 +72,7 @@ impl FromStr for Ip { } } -fn solve(input: &str) -> usize { +fn solve1(input: &str) -> usize { input.lines() .map(str::parse) .map(Result::unwrap) @@ -49,11 +80,20 @@ fn solve(input: &str) -> usize { .count() } +fn solve2(input: &str) -> usize { + input.lines() + .map(str::parse) + .map(Result::unwrap) + .filter(Ip::supports_ssl) + .count() +} + fn main() { let mut input = String::new(); io::stdin().read_to_string(&mut input).unwrap(); - println!("Part 1: {}", solve(&input)); + println!("Part 1: {}", solve1(&input)); + println!("Part 2: {}", solve2(&input)); } #[test] @@ -64,5 +104,16 @@ abcd[bddb]xyyx aaaa[qwer]tyui ioxxoj[asdfgh]zxcvbn "; - assert_eq!(2, solve(input.trim())); + assert_eq!(2, solve1(input.trim())); +} + +#[test] +fn part2() { + let input = " +aba[bab]xyz +xyx[xyx]xyx +aaa[kek]eke +zazbz[bzb]cdb +"; + assert_eq!(3, solve2(input.trim())); } |