From 1013a5f36d7be0d1c09b81b951eb4bbafd8ef828 Mon Sep 17 00:00:00 2001 From: Curtis McEnroe Date: Sat, 17 Dec 2016 01:25:18 -0500 Subject: Day 16 --- src/bin/day16.rs | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 50 insertions(+) create mode 100644 src/bin/day16.rs (limited to 'src/bin') diff --git a/src/bin/day16.rs b/src/bin/day16.rs new file mode 100644 index 0000000..21648b2 --- /dev/null +++ b/src/bin/day16.rs @@ -0,0 +1,50 @@ +use std::io::{self, Read}; + +fn fill_disk(initial_state: &str, len: usize) -> String { + let mut state = String::from(initial_state); + while state.len() < len { + let mut b: Vec = state.bytes() + .map(|b| { + match b { + b'0' => b'1', + b'1' => b'0', + _ => b, + } + }) + .collect(); + b.reverse(); + let b = String::from_utf8(b).unwrap(); + state.push('0'); + state.push_str(&b); + } + state.truncate(len); + state +} + +fn checksum(data: &str) -> String { + let mut sum = String::from(data); + while sum.len() % 2 == 0 { + sum = sum.as_bytes() + .chunks(2) + .map(|c| if c[0] == c[1] { '1' } else { '0' }) + .collect(); + } + sum +} + +fn solve(len: usize, initial_state: &str) -> String { + let data = fill_disk(initial_state, len); + checksum(&data) +} + +fn main() { + let mut input = String::new(); + io::stdin().read_to_string(&mut input).unwrap(); + + println!("Part 1: {}", solve(272, input.trim())); +} + +#[test] +fn part1() { + assert_eq!("01100", solve(20, "10000")); +} -- cgit 1.4.1