diff options
author | June McEnroe <june@causal.agency> | 2017-12-05 00:22:25 -0500 |
---|---|---|
committer | June McEnroe <june@causal.agency> | 2020-11-22 00:14:25 -0500 |
commit | 5aa7df1d78d7141c46b8d72f5788e36e330d837d (patch) | |
tree | 9a3306d14591d5a83352676505fe2e01c55e8071 /2017 | |
parent | Day 5 (diff) | |
download | aoc-5aa7df1d78d7141c46b8d72f5788e36e330d837d.tar.gz aoc-5aa7df1d78d7141c46b8d72f5788e36e330d837d.zip |
Day 5, part 2
Diffstat (limited to '')
-rw-r--r-- | 2017/src/bin/day05.rs | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/2017/src/bin/day05.rs b/2017/src/bin/day05.rs index b8e21d9..e96138e 100644 --- a/2017/src/bin/day05.rs +++ b/2017/src/bin/day05.rs @@ -18,14 +18,42 @@ fn solve1(input: &str) -> u32 { unreachable!() } +fn solve2(input: &str) -> u32 { + let mut jumps: Vec<isize> = input.lines() + .map(str::parse) + .map(Result::unwrap) + .collect(); + + let mut index = 0isize; + for step in 0.. { + if index < 0 || index >= jumps.len() as isize { + return step; + } + let jump = &mut jumps[index as usize]; + index += *jump; + if *jump >= 3 { + *jump -= 1; + } else { + *jump += 1; + } + } + unreachable!() +} + 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() { assert_eq!(5, solve1("0\n3\n0\n1\n-3\n")); } + +#[test] +fn part2() { + assert_eq!(10, solve2("0\n3\n0\n1\n-3\n")); +} |