开灯游戏（python）

题目描述
有9盏灯与9个开关，编号都是1~9。

每个开关能控制若干盏灯，按下一次会改变其控制的灯的状态(亮的变成不亮，不亮变成亮的)。

具体如下：

第一个开关控制第二，第四盏灯；

第二个开关控制第一，第三，第五盏灯；

第三个开关控制第二，第六盏灯；

第四个开关控制第一，第五，第七盏灯；

第五个开关控制第二，第四，第六，第八盏灯；

第六个开关控制第三，第五，第九盏灯；

第七个开关控制第四，第八盏灯；

第八个开关控制第五，第七，第九盏灯；

第九个开关控制第六，第八盏灯。

开始时所有灯都是熄灭的，开关是关闭着的。要求按下若干开关后，使得只有4盏灯亮着。
输入
无
输出
输出所有可能的方案，每行一个方案，每一行有9个字符，从左往右第i个字符表示第i个开关的状态(" 0" 表示关闭，" 1" 表示打开)，按字典序输出。下面的样例输出只是部分方案。
样例输入
无
样例输出
000001011
000001110
000001111

代码：

思路：直接暴力搜索所有情况
技巧：使用copy.deepcopy,使用异或提高运算速度。

import copy
control =   [[0,1,0,1,0,0,0,0,0],
            [1,0,1,0,1,0,0,0,0],
            [0,1,0,0,0,1,0,0,0],
            [1,0,0,0,1,0,1,0,0],
            [0,1,0,1,0,1,0,1,0],
            [0,0,1,0,1,0,0,0,1],
            [0,0,0,1,0,0,0,1,0],
            [0,0,0,0,1,0,1,0,1],
            [0,0,0,0,0,1,0,1,0]]

lights = [0 for _ in range(9)]
res = [0 for _ in range(9)]

# 考虑第lindex个开关的按情况
def dfs(lindex):
    global lights, res
    # 所有开关决策完毕
    if lindex == 9:
        if sum(lights) == 4:
            for x in res:
                print(x, end="")
            print()
        return
    # 所有开关未决策完毕
    else:
        #当前开关不动
        dfs(lindex+1)

        #当前开关按下
        lightsbackup = copy.deepcopy(lights)
        res[lindex] = 1
        for i in range(9):
            lights[i] = lights[i]^control[lindex][i]
        dfs(lindex+1)
        res[lindex] = 0
        lights = lightsbackup
        
dfs(0)