summary refs log tree commit diff homepage
path: root/2017/src/bin/day01.rs
blob: ee81d150161a6db659a1111dc56937db402f841c (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
use std::io::{self, Read};

fn solve(input: &str) -> u32 {
    let mut sum = 0;

    let chars = input.chars();
    let nexts = input.chars().cycle().skip(1);

    for (a, b) in chars.zip(nexts) {
        if a == b {
            sum += a.to_digit(10).unwrap();
        }
    }

    sum
}

fn main() {
    let mut input = String::new();
    io::stdin().read_to_string(&mut input).unwrap();

    println!("Part 1: {}", solve(input.trim()));
}

#[test]
fn part1() {
    assert_eq!(3, solve("1122"));
    assert_eq!(4, solve("1111"));
    assert_eq!(0, solve("1234"));
    assert_eq!(9, solve("91212129"));
}