From be489d9b3bfe62d53807e489910be0d1fd45ef98 Mon Sep 17 00:00:00 2001 From: Irene Knapp Date: Fri, 4 Dec 2020 21:57:15 -0800 Subject: 05, and some stray stuff from 04, oops --- 04/src/main.rs | 134 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 134 insertions(+) create mode 100644 04/src/main.rs (limited to '04/src') diff --git a/04/src/main.rs b/04/src/main.rs new file mode 100644 index 0000000..701f60d --- /dev/null +++ b/04/src/main.rs @@ -0,0 +1,134 @@ +use advent_lib::prelude::*; + +use regex::Regex; +use std::collections::BTreeSet; + + +fn main() -> Result<()> { + let mut args = std::env::args(); + if args.len() != 2 { + eprintln!("Usage: advent input"); + } + let _ = args.next(); + let filename = args.next().unwrap(); + + let input = advent_lib::read_lines_file(&filename)?; + + let overall_regex = Regex::new(r"([a-z][a-z][a-z]):([^ \n]*)").unwrap(); + let year_regex = Regex::new(r"^[0-9][0-9][0-9][0-9]$").unwrap(); + let height_regex = Regex::new(r"^([0-9]+)(cm|in)$").unwrap(); + let rgb_regex = Regex::new(r"^#[0-9a-f][0-9a-f][0-9a-f][0-9a-f][0-9a-f][0-9a-f]$").unwrap(); + let eye_regex = Regex::new(r"^(amb|blu|brn|gry|grn|hzl|oth)$").unwrap(); + let id_regex = Regex::new(r"^[0-9][0-9][0-9][0-9][0-9][0-9][0-9][0-9][0-9]$").unwrap(); + + let mut n_valid: i64 = 0; + let mut current_fields: BTreeSet = BTreeSet::new(); + let mut current_is_valid = true; + + for line in &input { + if line.trim().len() == 0 { + //println!("{:?}", current_fields); + + if current_is_valid + && current_fields.contains("byr") + && current_fields.contains("iyr") + && current_fields.contains("eyr") + && current_fields.contains("hgt") + && current_fields.contains("hcl") + && current_fields.contains("ecl") + && current_fields.contains("pid") { + n_valid += 1; + } + + current_fields = BTreeSet::new(); + current_is_valid = true; + } + + for captures in overall_regex.captures_iter(line) { + let mut field_name = String::new(); + field_name.push_str(&captures[1]); + + let mut field_value = String::new(); + field_value.push_str(&captures[2]); + + current_fields.insert(field_name); + + match &captures[1] { + "byr" => { + if year_regex.is_match(&field_value) { + let year = field_value.parse::()?; + if year < 1920 || year > 2002 { + current_is_valid = false; + } + } else { + current_is_valid = false; + } + }, + "iyr" => { + if year_regex.is_match(&field_value) { + let year = field_value.parse::()?; + if year < 2010 || year > 2020 { + current_is_valid = false; + } + } else { + current_is_valid = false; + } + }, + "eyr" => { + if year_regex.is_match(&field_value) { + let year = field_value.parse::()?; + if year < 2020 || year > 2030 { + current_is_valid = false; + } + } else { + current_is_valid = false; + } + }, + "hgt" => { + if height_regex.is_match(&field_value) { + let captures = height_regex.captures(&field_value).unwrap(); + let number = captures[1].parse::()?; + + match &captures[2] { + "cm" => { + if number < 150 || number > 193 { + current_is_valid = false; + } + }, + "in" => { + if number < 59 || number > 76 { + current_is_valid = false; + } + }, + _ => { }, + } + } else { + current_is_valid = false; + } + }, + "hcl" => { + if !rgb_regex.is_match(&field_value) { + current_is_valid = false; + } + }, + "ecl" => { + if !eye_regex.is_match(&field_value) { + current_is_valid = false; + } + }, + "pid" => { + if !id_regex.is_match(&field_value) { + current_is_valid = false; + } + }, + _ => { + }, + }; + } + //println!("end of line"); + } + + println!("{}", n_valid); + + Ok(()) +} -- cgit 1.4.1