diff --git a/day10/day10.py b/day10/day10.py index ac8d8ee..fcf67c1 100644 --- a/day10/day10.py +++ b/day10/day10.py @@ -16,9 +16,34 @@ def part1(): print("Part 1 %d " % (diffs[1] * diffs[3])) +def count_options(og, adapters): + amount = 0 + adap = list.copy(adapters) + + if len(adap) == 1: + return 1 + + while True: + num = adap.pop(0) + if num > og + 3: + break + amount += count_options(num, adap) + + return amount + + def part2(): file = open("input.txt") - lines = [int(i) for i in file.readlines()] + + adapters = [0] + [adapters.append(int(i)) for i in file.readlines()] + adapters.append(max(adapters) + 3) + + adapters = sorted(adapters) + og = adapters.pop(0) + ans = count_options(og, adapters) + + print("Part 2: %d" % ans) if __name__ == "__main__":