diff options
Diffstat (limited to '2023/day01.rs')
| -rw-r--r-- | 2023/day01.rs | 55 |
1 files changed, 55 insertions, 0 deletions
diff --git a/2023/day01.rs b/2023/day01.rs new file mode 100644 index 0000000..3a3bfb2 --- /dev/null +++ b/2023/day01.rs @@ -0,0 +1,55 @@ +#![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; + for line in &lines { + let nums: Vec<_> = line.chars().filter_map(|a| a.to_digit(10)).collect(); + if nums.len() == 1 { + res1 += nums[0] * 10 + nums[0]; + } else { + res1 += nums[0] * 10 + nums[nums.len() - 1]; + } + } + + let mut res2 = 0; + let digits = [ + "zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine", + ]; + for line in &lines { + let b: Vec<char> = line.chars().collect(); + let mut nums: Vec<u32> = Vec::new(); + for i in 0..b.len() { + match b[i].to_digit(10) { + Some(i) => nums.push(i), + None => (), + } + for (j, digit) in digits.iter().enumerate() { + let d: Vec<char> = digit.chars().collect(); + if d.iter().zip(b[i..].iter()).filter(|&(a, b)| a == b).count() == d.len() { + nums.push(j as u32); + } + } + } + if nums.len() == 0 { + res2 += nums[0] * 10 + nums[0]; + } else { + res2 += nums[0] * 10 + nums[nums.len() - 1]; + } + } + + println!("res1: {}", res1); + println!("res2: {}", res2); +} |
