blob: 91b90ff00fedc5ae9bf0662145f8046c239fcb34 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
from collections import deque
from collections.abc import Iterable, Iterator, Generator
from itertools import islice
from sys import argv
from typing import TypeVar
T = TypeVar('T')
def sliding_window(iterable: Iterable[T], n: int) -> Generator[tuple[T, ...], None, None]:
# sliding_window('ABCDEFG', 4) -> ABCD BCDE CDEF DEFG
it: Iterator[T] = iter(iterable)
window: deque[T] = deque(islice(it, n), maxlen=n)
if len(window) == n:
yield tuple(window)
for x in it:
window.append(x)
yield tuple(window)
def open_day(n: int):
if len(argv) == 2:
return open(argv[1])
return open(f'{n}.in')
|
