mirror of
https://github.com/Crocmagnon/advent-of-code.git
synced 2024-11-16 03:34:00 +01:00
58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
from collections.abc import Generator
|
|
from typing import TypeVar
|
|
|
|
|
|
def main(filename: str, expected_part_1: int = None, expected_part_2: int = None):
|
|
print(f"\n+ Running on {filename}")
|
|
with open(filename) as f:
|
|
data = f.read().strip().split("\n")
|
|
|
|
data = parse_data(data)
|
|
solution_part_1 = solve_part_1(data)
|
|
|
|
print(f"1. Found {solution_part_1}")
|
|
if expected_part_1:
|
|
assert expected_part_1 == solution_part_1
|
|
|
|
solution_part_2 = solve_part_2(data)
|
|
print(f"2. Found {solution_part_2}")
|
|
if expected_part_2:
|
|
assert expected_part_2 == solution_part_2
|
|
|
|
|
|
DataType = str
|
|
T = TypeVar("T")
|
|
|
|
|
|
def parse_data(data: list[str]) -> DataType:
|
|
return data[0]
|
|
|
|
|
|
def solve_part_1(data: DataType) -> int:
|
|
size = 4
|
|
for index, window in enumerate(sliding_window(data, size)):
|
|
if not repeating(window):
|
|
return index + size
|
|
return 0
|
|
|
|
|
|
def repeating(chars: str) -> bool:
|
|
return len(set(chars)) != len(chars)
|
|
|
|
|
|
def sliding_window(stream: str, size: int) -> Generator[str]:
|
|
for i in range(len(stream) - size):
|
|
yield stream[i : i + size]
|
|
|
|
|
|
def solve_part_2(data: DataType) -> int:
|
|
size = 14
|
|
for index, window in enumerate(sliding_window(data, size)):
|
|
if not repeating(window):
|
|
return index + size
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main("inputs/day06-test1", expected_part_1=10, expected_part_2=29)
|
|
main("inputs/day06", expected_part_1=1578, expected_part_2=2178)
|