diff --git a/src/tools/aoc.py b/src/tools/aoc.py index 06e0142..69e6f85 100644 --- a/src/tools/aoc.py +++ b/src/tools/aoc.py @@ -207,6 +207,12 @@ class AOCDay: else: return self.input.copy() + def getIntsFromInput(self) -> list: + if len(self.input) == 1: + return list(map(int, re.findall(r"-?\d+", self.input[0]))) + else: + return [list(map(int, re.findall(r"-?\d+", l))) for l in self.input] + def getMultiLineInputAsArray( self, return_type: Type = None, join_char: str = None ) -> List: