aboutsummaryrefslogtreecommitdiff
path: root/src/day5pt2.rs
blob: 0d92b5df2cebedd8306f6ac8c63be55bbea1f894 (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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
use std::cmp;

pub fn main() {
    let txt = std::fs::read_to_string("./input/day5.txt").unwrap();

    let mut seeds: Vec<(i64, i64)> = vec![];
    let mut rules: Vec<(i64, i64, i64)> = vec![];

    for (i, line) in txt.lines().enumerate() {
        if line != "" {
            // dodaj u vec
            if i != 0 && line.chars().nth(line.len() - 1).unwrap() != ':' {
                let br: Vec<i64> = line.split(" ").map(|s| s.parse().unwrap()).collect();
                rules.push((br[0], br[1], br[2] - 1));
            }
        } else {
            // obradi
            let mut obradjeno: Vec<(i64, i64)> = vec![];
            let mut neobradjeno: Vec<(i64, i64)> = vec![];

            neobradjeno.clear();
            neobradjeno.extend(seeds.clone());

            while neobradjeno.len() != 0 {
                seeds.clear();
                seeds.extend(neobradjeno.clone());
                neobradjeno.clear();

                for seed in &seeds {
                    let mut ind: bool = false;
                    for rule in &rules {
                        let l_seed = seed.0;
                        let r_seed = seed.1;
                        let l_rule = rule.1;
                        let r_rule = rule.1 + rule.2;

                        if r_rule < l_seed || r_seed < l_rule {
                            continue;
                        }
                        ind = true;
                        let levi = cmp::max(l_seed, l_rule);
                        let desni = cmp::min(r_seed, r_rule);
                        obradjeno.push((levi + rule.0 - rule.1, desni + rule.0 - rule.1));

                        if l_seed != levi {
                            neobradjeno.push((l_seed, levi - 1));
                        }

                        if desni != r_seed {
                            neobradjeno.push((desni + 1, r_seed));
                        }
                    }

                    if !ind {
                        obradjeno.push((seed.0, seed.1));
                    }
                }
            }

            seeds = obradjeno;
            rules.clear();
        }

        if seeds.is_empty() && line != "" {
            let x: Vec<i64> = line
                .split(": ")
                .collect::<Vec<_>>()
                .get(1)
                .unwrap()
                .split(" ")
                .map(|s| s.parse().unwrap())
                .collect();

            let mut j = 0;
            while j < x.len() {
                seeds.push((x[j], x[j] + x[j + 1] - 1));
                j += 2;
            }
        }
    }

    let mut min = i64::MAX;
    for seed in seeds {
        if seed.0 < min {
            min = seed.0;
        }
    }

    println!("{}", min);
}