aboutsummaryrefslogtreecommitdiff
path: root/src/day1pt2.rs
blob: 99407dd1865bd5d027e019d8c3ff237c5a4a8849 (plain)
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
use regex::Regex;

fn parse(s: &str) -> u32 {
    match s {
        "zero" => 0,
        "one" => 1,
        "two" => 2,
        "three" => 3,
        "four" => 4,
        "five" => 5,
        "six" => 6,
        "seven" => 7,
        "eight" => 8,
        "nine" => 9,
        br => br.parse().unwrap(),
    }
}

fn parse_bw(s: &str) -> u32 {
    match s {
        "orez" => 0,
        "eno" => 1,
        "owt" => 2,
        "eerht" => 3,
        "ruof" => 4,
        "evif" => 5,
        "xis" => 6,
        "neves" => 7,
        "thgie" => 8,
        "enin" => 9,
        br => br.parse().unwrap(),
    }
}

pub fn main() {
    let mut sum: u32 = 0;
    let re = Regex::new(r"\d|zero|one|two|three|four|five|six|seven|eight|nine").unwrap();
    let re_bw = Regex::new(r"\d|ozer|eno|owt|eerht|ruof|evif|xis|neves|thgie|enin").unwrap();

    let txt = std::fs::read_to_string("./input/day1.txt").unwrap();
    for line in txt.lines() {
        let capt = re.find_iter(line).collect::<Vec<_>>();
        let str_bw = String::from_iter(line.chars().rev());
        let capt_bw = re_bw.find_iter(&str_bw).collect::<Vec<_>>();

        let x =
            parse(capt.get(0).unwrap().as_str()) * 10 + parse_bw(capt_bw.get(0).unwrap().as_str());
        sum += x;
    }
    println!("{}", sum)
}