广度优先法速通浅塘

import keyboard, pyautogui, sys, time
from PIL import ImageDraw
X, Y, gridSize, puzzle = 77, 306    , 84, ""  # (X,Y)为左上格中心坐标，gridsize为格子大小
answer, checkers, states, test = "", "", set(), True
def dragTo(s, goal):
    global X, Y
    pyautogui.mouseDown()
    pyautogui.moveTo(X+(s%6+goal)*gridSize, Y+(s//6)*gridSize, duration=0.2)
    pyautogui.moveTo(X+(s%6+goal)*gridSize, Y+(s//6)*gridSize, duration=0.2)
    pyautogui.mouseUp()
def move(s):
    global puzzle
    p = puzzle
    for i in range(0, len(s), 2):
        checker, d = s[i: i + 2]
        first, last = p.find(checker), p.rfind(checker)
        if d == '↑': p = p[:first - 6] + checker + p[first - 5:last] + "_" + p[last + 1:]
        if d == '↓': p = p[:first] + "_" + p[first + 1:last + 6] + checker + p[last + 7:]
        if d == '←': p = p[:first - 1] + checker + p[first:last] + "_" + p[last + 1:]
        if d == '→': p = p[:first] + "_" + p[first + 1:last + 1] + checker + p[last + 2:]
    return p
def moves(s):
    n = 0
    for i in range(0, len(s), 2):
        n += (i == 0 or s[i - 2:i] != s[i:i + 2])
    return n
while True:
    keyboard.on_press_key("alt", lambda x: sys.exit())
    screen = pyautogui.screenshot()
    if test:
        screenTest = pyautogui.screenshot()
        draw = ImageDraw.Draw(screenTest)
    pic, row, col, puzzle = "", ord("a"), ord("A"), ""
    for i in range(121):
        xx, yy = X + (i % 11) * gridSize / 2, Y + (i // 11) * gridSize / 2
        kx = gridSize/2-2 if (i % 11) % 2 == 0 else 2
        ky = gridSize/2-2 if (i // 11) % 2 == 0 else 2
        rect_area = screen.crop((xx - kx, yy - ky, xx + kx, yy + ky)).resize((1, 1))
        draw.rectangle([(xx - kx, yy - ky), (xx + kx, yy + ky)], fill=None, outline="red")
        color = rect_area.getpixel((0, 0))
        if color[0] < 150:
            pic += "蓝" if color[1] < 160 else "口"
        else:
            pic += "红" if color[1] < 150 else "黄"
    if test:
        screenTest.save("D:/录像/封面.jpg")
        test = False
    # for i in range(11): print(pic[i*11:(i+1)*11])
    for i in range(36):
        x, y = i // 6, i % 6  # x是行数，y是列数
        xy = x * 2 * 11 + y * 2
        color = pic[xy]
        if color == "红":
            puzzle += "x"
        elif color == "口":
            puzzle += "_"
        elif (x == 0 or pic[xy - 11] == "口") and (x == 5 or pic[xy + 11] == "口"):  # 行块
            puzzle += chr(row)
            row += (y == 5 or pic[xy + 1] == "口")
        else:  # 列块
            if x > 0 and pic[xy - 11] != "口":
                puzzle += puzzle[(x - 1) * 6 + y]
            else:
                puzzle += chr(col)
                col += 1
    checkers = set(''.join(i for i in puzzle.split()))
    answer, states, steps = "", set(), [""]
    while not answer:
        steps0 = []
        for step in steps:
            p = move(step)
            if p not in states:
                states.add(p)
            else:
                continue
            if p.rfind('x') % 6 == 5: answer = step; break
            for checker in checkers:
                first, last = p.find(checker), p.rfind(checker)
                if checker < 'a':  # 竖子
                    i = 1
                    while first - i * 6 >= 0 and p[first - i*6] == "_":
                        steps0 += [step + (checker + "↑") * i]
                        i += 1
                    i = 1
                    while last + i * 6 < 36 and p[last + i*6] == "_":
                        steps0 += [step + (checker + "↓") * i]
                        i += 1
                else:  # 横子
                    i = 1
                    while first//6 == (first - i)//6 and p[first - i] == "_":
                        steps0 += [step + (checker + "←") * i]
                        i += 1
                    i = 1
                    while last//6 == (last + i)//6 and p[last + i] == "_":
                        steps0 += [step + (checker + "→") * i]
                        i += 1
        if not answer: steps = [i for i in steps0]
        print(len(steps))
    print("步骤", answer)
    print("步数", moves(answer))
    i = 0
    while i < len(answer):
        p = move(answer[:i])
        checker, d = answer[i: i + 2]
        n, first, last = 1, p.find(checker), p.rfind(checker)
        while i + 2 < len(answer) and answer[i + 2] == answer[i]:
            i, n = i + 2, n + 1
        else:
            i += 2
        if d in ['↑', '←']:
            pyautogui.moveTo(X + (first % 6) * gridSize, Y + (first // 6) * gridSize)
            dragTo(first - n * (6 if d == '↑' else 1), i == len(answer))
        else:
            pyautogui.moveTo(X + (last % 6) * gridSize, Y + (last // 6) * gridSize)
            dragTo(last + n * (6 if d == '↓' else 1), i == len(answer))
    time.sleep(0.8)
    for i in range(2): pyautogui.click(450, 400)
    for i in range(3): pyautogui.click(300, 666)
    time.sleep(2)