from itertools import count

from aocd.models import Puzzle
from aocd import submit


def get_marker_pos(signal, N):
    """
    iterate over input stream and find all the first occurence
    where the Set from i to i+N equals N. This means there are
    no duplicates within this particular range.
    """
    for i in count():
        if len(set(signal[i : i + N])) == N:
            return i + N


if __name__ == "__main__":
    # get puzzle and parse data
    puzzle = Puzzle(year=2022, day=6)
    signal = puzzle.input_data

    # part a:
    answer_a = get_marker_pos(signal, 4)
    print(f"{answer_a}")
    submit(answer_a, part="a", day=6, year=2022)

    # part b:
    answer_b = get_marker_pos(signal, 14)
    print(f"{answer_b}")
    submit(answer_b, part="b", day=6, year=2022)