summaryrefslogtreecommitdiff
path: root/day10/src/lib.rs
blob: f4834207785fbc3a2cccaee72cd0d3a6705bf831 (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
91
92
93
94
95
96
97
98
99
100
101
102
103
use std::collections::VecDeque;
use std::str::FromStr;

pub struct SyntaxLines {
    lines: Vec<String>,
}

impl FromStr for SyntaxLines {
    type Err = ();

    fn from_str(input: &str) -> Result<Self, Self::Err> {
        let lines: Vec<_> = input
            .trim()
            .split('\n')
            .map(|x| x.to_owned())
            .collect::<Vec<String>>();

        Ok(SyntaxLines { lines })
    }
}

impl SyntaxLines {
    pub fn part1(&self) -> u32 {
        let mut result: u32 = 0;

        for line in &self.lines {
            let mut qopen = VecDeque::new();
            let mut panic_char = ' ';

            for character in line.chars() {
                match character {
                    c if character == '('
                        || character == '['
                        || character == '{'
                        || character == '<' =>
                    {
                        qopen.push_back(c);
                    }
                    c if character == ')'
                        || character == ']'
                        || character == '}'
                        || character == '>' =>
                    {
                        match qopen.pop_back() {
                            Some(bracket) => {
                                let mut error = false;
                                if c == ')' && bracket != '(' {
                                    error = true;
                                } else if c == ']' && bracket != '[' {
                                    error = true;
                                } else if c == '}' && bracket != '{' {
                                    error = true;
                                } else if c == '>' && bracket != '<' {
                                    error = true;
                                }

                                if error {
                                    panic_char = c;
                                }
                            }
                            None => panic!("Wtf?"),
                        };
                    }
                    _ => {}
                };

                if panic_char != ' ' {
                    if panic_char == ')' {
                        result += 3;
                    } else if panic_char == ']' {
                        result += 57;
                    } else if panic_char == '}' {
                        result += 1197;
                    } else if panic_char == '>' {
                        result += 25137;
                    }
                    break;
                }
            }
        }

        result
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_example() {
        let input: SyntaxLines = include_str!("../example.txt").parse().unwrap();
        let result = input.part1();
        assert_eq!(result, 26397);
    }

    #[test]
    fn test_puzzle_input() {
        let input: SyntaxLines = include_str!("../input.txt").parse().unwrap();
        let result = input.part1();
        assert_eq!(result, 392367);
    }
}