import intsets, math, sequtils, strutils

let input = readFile("data/01.txt").strip.splitLines.map(parseInt)

func part1(freqs: seq[int]): int =
    freqs.sum

func part2(freqs: seq[int]): int =
    var
        seen = initIntSet()
        talley: int

    while true:
        for n in freqs:
            talley += n
            if talley in seen:
                return talley
            seen.incl(talley)

echo part1(input)
echo part2(input)