diff --git a/sixx.py b/sixx.py index f7f9885..bcedba2 100755 --- a/sixx.py +++ b/sixx.py @@ -1,11 +1,11 @@ #!/usr/bin/env python3 from useful import * -ls = lines(open(0)) -def n(l): +def n(l, x): for i, _ in enumerate(l): - if len(list(set(l[i:i+4]))) == 4: - return i+4 + if len(list(set(l[i:i+x]))) == x: + return i+x -ic([n(l) for l in ls]) -print(sum([n(l) for l in ls])) +ls = lines(open(0)) +print(sum([n(l, 4) for l in ls])) +print(sum([n(l, 14) for l in ls])) diff --git a/test6 b/test6 new file mode 100644 index 0000000..7980a82 --- /dev/null +++ b/test6 @@ -0,0 +1 @@ +mjqjpqmgbljsphdztnvjfqwrcgsmlb