'optimize' day 6

This commit is contained in:
Michal Kunc 2022-12-06 09:59:07 +01:00
parent 314b38e25c
commit 389e5faed2

View File

@ -2,18 +2,13 @@
import fileinput import fileinput
from typing import Iterable from typing import Iterable
def solve(data: str) -> int: def solve(data: str, length: int = 4) -> int:
for idx in range(len(data)): for idx in range(len(data)):
if len(set(list(data[idx:idx+4]))) == 4: if len(set(list(data[idx:idx+length]))) == length:
return idx + 4 return idx + length
def solveb(data: str) -> int:
for idx in range(len(data)):
if len(set(list(data[idx:idx+14]))) == 14:
return idx + 14
if __name__ == "__main__": if __name__ == "__main__":
with fileinput.input() as f: with fileinput.input() as f:
data = f.readline() data = f.readline()
print(solve(data)) print(solve(data, 4))
print(solveb(data)) print(solve(data, 14))