aoc

Advent of Code
git clone https://git.pngu.org/aoc
Log | Files | Refs | README

6_2022.py (434B)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
import sys

infile = sys.argv[1] if len(sys.argv) > 1 else "6_2022.txt"
data = open(infile).read()
lines = [x for x in data.split("\n")]

p1 = False
p2 = False
for i in range(len(data)):
    if (not p1) and i - 3 >= 0 and len(set([data[i - j] for j in range(4)])) == 4:
        print(i + 1)
        p1 = True
    if (not p2) and i - 13 >= 0 and len(set([data[i - j] for j in range(14)])) == 14:
        print(i + 1)
        p2 = True