DAY19

gistfile1.txt

from termcolor import cprint


SAMPLES = """
   5
"""


def part_one(input):
    cprint("Day 19: part one ", 'green')

    # input = SAMPLES
    party_size = 0

    elves = {}

    for line in input.splitlines():
        if not line:
            continue
        party_size = int(line)

    for elf in range(1, party_size+1):
        elves[elf] = 1

    print white_elephant(elves)


def white_elephant(elves):

    while True:
        something_stolen = False

        for elf in elves:
            if elves[elf] > 0:
                for i in range(1, len(elves)-1):
                    vitim_elf = ((elf+i) % len(elves)) or len(elves)
                    if elves[vitim_elf] > 0:
                        elves[elf] += elves[vitim_elf]
                        elves[vitim_elf] = 0
                        something_stolen = True
                        break

        if not something_stolen:
            break

    for elf in elves:
        if elves[elf] > 0:
            print elf


def part_two(input):
    cprint("Day 19: part two ", 'green')

    # for line in input.splitlines():
    #     if not line:
    #         continue
    #     party_size = int(line)