hex.py

import sys


def equal(triplet):
    return len(set(triplet)) == 1


def finished(world):
    for i in range(len(world)):
        if len(world[i:]) < 3:
            return True
        triplet = world[i:i + 3]
        if equal(triplet):
            return False


def collapse(world):
    print(world)

    i = 0
    while not finished(world):
        triplet = world[i:i + 3]
        if equal(triplet):
            world.insert(i, sum([world.pop(i), world.pop(i), world.pop(i)]))
            print(world)
            i = 0
        else:
            i += 1

if __name__ == '__main__':
    num = int(sys.argv[1])
    world = [1 for i in range(num)]
    collapse(world)