import itertools

# -------------------------------#
# ========== SETTINGS ========== #
# -------------------------------#

# Note: initial positions are counted clockwise from the bottom for the left wheel/console,
# and anti-clockwise from the bottom for the right wheel/console, zero-indexed.

wheel_size_left = 6  # default = 6
wheel_size_right = 6  # default = 6

start_wheel = 0  # default = 0
start_position = 0  # default = 0
end_wheel = 1  # default = 1
end_position = 4  # default = 4

valid_positions_left = {0, 2, 4}  # default = {0, 2, 4}
valid_positions_right = {1, 3, 5}  # default = {1, 3, 5}

jumps_left_to_right = {0: 0, 3: 3, 4: 4, 5: 5}  # default = {0: 0, 3: 3, 4: 4, 5: 5}
jumps_right_to_left = {0: 0, 3: 3, 4: 4, 5: 5}  # default = {0: 0, 3: 3, 4: 4, 5: 5}

blocked_pegs_left = {0, 2}  # default = {0, 2}
blocked_pegs_right = {0}  # default = {0}

# -------------------------------#
# ========== INTERNAL ========== #
# -------------------------------#

available_pegs_left = set(range(wheel_size_left)) - blocked_pegs_left
available_pegs_right = set(range(wheel_size_right)) - blocked_pegs_right


class Wheel:
    def __init__(self, size, raw_valid_positions: set):
        self.size = size
        self.raw_valid_positions = raw_valid_positions
        self.rotation = 0

    def rotate(self):
        self.rotation += 1
        self.rotation %= self.size

    def is_valid_position(self, position):
        position += (self.size - self.rotation)
        return position % self.size in self.raw_valid_positions


class Console:
    def __init__(self, size, pegs: set):
        self.size = size
        self.pegs = pegs
        self.current_position = 0

    def increment_position(self):
        self.current_position += 1
        self.current_position %= self.size


class Puzzle:
    def __init__(self, wheel_left: Wheel, wheel_right: Wheel, console_left: Console, console_right: Console):
        self.wheel_left = wheel_left
        self.wheel_right = wheel_right
        self.console_left = console_left
        self.console_right = console_right
        self.ball_wheel = start_wheel
        self.ball_position = start_position

    def current_wheel(self):
        return self.wheel_left if self.ball_wheel == 0 else self.wheel_right

    def current_console(self):
        return self.console_left if self.ball_wheel == 0 else self.console_right

    def is_fail(self):
        wheel = self.current_wheel()
        if not wheel.is_valid_position(self.ball_position):
            return True
        console = self.current_console()
        if len(console.pegs) == 0 or max(console.pegs) < console.current_position:
            return True
        return False

    def move(self):
        wheel = self.current_wheel()
        wheel.rotate()
        self.ball_position += (wheel.size - 1)
        self.ball_position %= wheel.size
        self.current_console().increment_position()

    def get_jumps(self) -> dict:
        return jumps_left_to_right if self.ball_wheel == 0 else jumps_right_to_left

    def try_jump(self):
        console = self.current_console()
        if console.current_position in console.pegs:
            jumps = self.get_jumps()
            if self.ball_position in jumps:
                self.ball_position = jumps[self.ball_position]
                self.ball_wheel = 1 - self.ball_wheel
                console.pegs.remove(console.current_position)
                wheel = self.current_wheel()
                if not wheel.is_valid_position(self.ball_position):
                    if self.ball_wheel == end_wheel and self.ball_position == end_position:
                        return 1
                    return -1
            else:
                return -1
        return 0

    def simulate(self):
        while True:
            if self.is_fail():
                return False
            self.move()
            jump = self.try_jump()
            if jump == 1:
                return len(self.console_left.pegs) == 0 and len(self.console_right.pegs) == 0
            if jump == -1:
                return False


def subsets(s, n):
    return map(lambda x: {} if not x else set(x), itertools.combinations(s, n))


def test(pegs_left, pegs_right):
    wheel_left = Wheel(wheel_size_left, valid_positions_left)
    wheel_right = Wheel(wheel_size_right, valid_positions_right)
    console_left = Console(wheel_size_left, pegs_left)
    console_right = Console(wheel_size_right, pegs_right)
    return Puzzle(wheel_left, wheel_right, console_left, console_right).simulate()


for i in range(0, 1 + len(available_pegs_left)):
    for j in range(0, 1 + len(available_pegs_right)):
        for left in subsets(available_pegs_left, i):
            for right in subsets(available_pegs_right, j):
                if test(left.copy(), right.copy()):
                    print(list(left), list(right))