lightsout.py 文件源码

def successors (config):
    import math
    leds = len(config)
    size = int(math.sqrt(leds))
    succs = []
    for i in range(leds):
        y = i // size
        x = i % size
        succ = np.copy(config)
        succ[i] *= -1
        if x-1 >= 0:
            succ[i-1] *= -1
        if x+1 < size:
            succ[i+1] *= -1
        if y-1 >= 0:
            succ[i-size] *= -1
        if y+1 < size:
            succ[i+size] *= -1
        succs.append(succ)
    return succs