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.
advent-of-code/2017/rust/src/day_01.rs

33 lines
854 B

use failure::*;
#[allow(dead_code)]
pub fn solve(input: &str) -> Result<String, Error> {
let input: Vec<u32> = input
.trim()
.chars()
.map(|x| x.to_digit(10).ok_or_else(|| err_msg("")))
.collect::<Result<_, _>>()?;
// 1
// let offset = 1;
let offset = input.len() / 2;
let offset_iter = input.iter().cycle().skip(offset);
Ok(input
.iter()
.zip(offset_iter)
.filter(|&a| a.0 == a.1)
.map(|a| a.0)
.sum::<u32>()
.to_string())
}
#[test]
fn part_two() {
assert_eq!(solve("1212").unwrap(), "6".to_string());
assert_eq!(solve("1221").unwrap(), "0".to_string());
assert_eq!(solve("123425").unwrap(), "4".to_string());
assert_eq!(solve("123123").unwrap(), "12".to_string());
assert_eq!(solve("12131415").unwrap(), "4".to_string());
}