summaryrefslogtreecommitdiff
path: root/2022/Rust/src/day4.rs
blob: 0dbc3dc2a120fc25207d50cdd5c0a249d994d46a (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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
use std::{num::ParseIntError, str::FromStr};

#[derive(Debug)]
struct Rng {
    start: usize,
    end: usize,
}

impl Rng {
    fn intersects(&self, rng: &Rng) -> bool {
        if self.start <= rng.end && rng.start <= self.end {
            return true;
        }
        false
    }

    fn contains(&self, rng: &Rng) -> bool {
        if self.start >= rng.start && self.end <= rng.end {
            true
        } else {
            false
        }
    }
}

impl FromStr for Rng {
    type Err = ParseIntError;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        let (start, end) = s.split_once('-').unwrap();
        let start_fromstr = start.parse::<usize>()?;
        let end_fromstr = end.parse::<usize>()?;
        Ok(Rng {
            start: start_fromstr,
            end: end_fromstr,
        })
    }
}

fn solve_part_one(data: &str) -> u32 {
    data.lines()
        .map(|line| -> u32 {
            let l = line.split_once(',').unwrap();
            let r1 = Rng::from_str(l.0).unwrap();
            let r2 = Rng::from_str(l.1).unwrap();
            (r1.contains(&r2) || r2.contains(&r1)) as u32
        })
        .sum()
}

fn solve_part_two(data: &str) -> u32 {
    data.lines()
        .map(|line| -> (Rng, Rng) {
            let l = line.split_once(',').unwrap();
            (Rng::from_str(l.0).unwrap(), Rng::from_str(l.1).unwrap())
        })
        .map(|rngs| rngs.0.intersects(&rngs.1) as u32)
        .sum()
}

fn main() {
    let data_test = include_str!("../input/day4.test");
    let data_prod = include_str!("../input/day4.prod");
    println!("{}", solve_part_one(data_test));
    println!("{}", solve_part_two(data_test));
    println!("{}", solve_part_one(data_prod));
    println!("{}", solve_part_two(data_prod));
}