You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
660 B
26 lines
660 B
7 years ago
|
use std::collections::HashMap;
|
||
|
|
use std::collections::hash_map::Entry;
|
||
|
|
use failure::*;
|
||
|
|
|
||
|
|
pub fn solve(input: &str) -> Result<String, Error> {
|
||
|
|
let mut jumps: HashMap<isize, isize> = input
|
||
|
|
.trim()
|
||
|
|
.lines()
|
||
|
|
.enumerate()
|
||
|
|
.map(|(i,x)| x.parse().map(|x| (i as isize, x)))
|
||
|
|
.collect::<Result<_,_>>()?;
|
||
|
|
let mut pc = 0;
|
||
|
|
let mut steps = 0;
|
||
|
|
while let Entry::Occupied(mut entry) = jumps.entry(pc) {
|
||
|
|
let offset = entry.get_mut();
|
||
|
|
pc += *offset;
|
||
|
|
if *offset >= 3 {
|
||
|
|
*offset -= 1;
|
||
|
|
} else {
|
||
|
|
*offset += 1;
|
||
|
|
}
|
||
|
|
steps += 1;
|
||
|
|
}
|
||
|
|
Ok(steps.to_string())
|
||
|
|
}
|