#![allow(dead_code)]
#![allow(unused_variables)]
#![allow(unused_mut)]
use std::fs::File;
use std::io::Read;

fn main() {
    // let filename = "in/day01.ref";
    let filename = "in/day01.pzl";

    let mut f = File::open(filename).expect("cannot open file");
    let mut content = String::new();
    f.read_to_string(&mut content).expect("cannot read file");
    let lines: Vec<_> = content.trim_end().split('\n').collect();
    // println!("{:?}", lines);

    let mut res1 = 0;
    let mut res2 = 0;

    let digits = [
        "zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine",
    ];

    for line in &lines {
        let mut nums1: Vec<u32> = Vec::new();
        let mut nums2: Vec<u32> = Vec::new();
        for (i, c) in line.chars().enumerate() {
            if let Some(n) = c.to_digit(10) {
                nums1.push(n);
                nums2.push(n);
            }
            for (d, d_str) in digits.iter().enumerate() {
                if line[i..].starts_with(d_str) {
                    nums2.push(d as u32);
                }
            }
        }
        res1 += nums1[0] * 10 + nums1[nums1.len() - 1];
        res2 += nums2[0] * 10 + nums2[nums2.len() - 1];
    }

    println!("res1: {}", res1);
    println!("res2: {}", res2);
    assert_eq!(res1, 55538);
    assert_eq!(res2, 54875);
}