'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
from typing import Iterable
def solve(data: str) -> int:
def solve(data: str, length: int = 4) -> int:
for idx in range(len(data)):
if len(set(list(data[idx:idx+4]))) == 4:
return idx + 4
def solveb(data: str) -> int:
for idx in range(len(data)):
if len(set(list(data[idx:idx+14]))) == 14:
return idx + 14
if len(set(list(data[idx:idx+length]))) == length:
return idx + length
if __name__ == "__main__":
with fileinput.input() as f:
data = f.readline()
print(solve(data))
print(solveb(data))
print(solve(data, 4))
print(solve(data, 14))