From 051be932a389b8bc3ea5d4626575454844639066 Mon Sep 17 00:00:00 2001 From: June McEnroe Date: Mon, 27 Nov 2017 17:11:18 -0500 Subject: Move to 2016 directory --- 2016/src/bin/day20.rs | 77 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 77 insertions(+) create mode 100644 2016/src/bin/day20.rs (limited to '2016/src/bin/day20.rs') diff --git a/2016/src/bin/day20.rs b/2016/src/bin/day20.rs new file mode 100644 index 0000000..0416ced --- /dev/null +++ b/2016/src/bin/day20.rs @@ -0,0 +1,77 @@ +use std::io::{self, Read}; +use std::u32; + +fn parse_ranges(input: &str) -> Vec<(u32, u32)> { + let mut ranges = vec![]; + for line in input.lines() { + let hyphen = line.find('-').unwrap(); + let (start, end) = line.split_at(hyphen); + let start = start.parse().unwrap(); + let end = end[1..].parse().unwrap(); + ranges.push((start, end)); + } + ranges.sort(); + ranges +} + +fn solve1(input: &str) -> u32 { + let ranges = parse_ranges(input); + let mut lowest = 0; + + for &(start, end) in &ranges { + if lowest >= start && lowest <= end { + lowest = end + 1; + } + } + + lowest +} + +fn solve2(input: &str) -> u32 { + let ranges = parse_ranges(input); + let mut merged = vec![]; + + for &(start, end) in &ranges { + if let Some(&mut (ref mut last_start, ref mut last_end)) = merged.last_mut() { + if start >= *last_start && end <= *last_end { + continue; + } else if start >= *last_start && start <= *last_end + 1 && end > *last_end { + *last_end = end; + continue; + } + } + merged.push((start, end)); + } + + let blocked: u32 = merged.iter().map(|&(start, end)| end - start + 1).sum(); + + u32::MAX - blocked + 1 +} + +fn main() { + let mut input = String::new(); + io::stdin().read_to_string(&mut input).unwrap(); + + println!("Part 1: {}", solve1(&input)); + println!("Part 2: {}", solve2(&input)); +} + +#[test] +fn part1() { + let input = " +5-8 +0-2 +4-7 +"; + assert_eq!(3, solve1(input.trim())); +} + +#[test] +fn part2() { + let input = " +5-8 +0-2 +4-7 +"; + assert_eq!(u32::MAX - 7, solve2(input.trim())); +} -- cgit 1.4.1