def get_line_remind(_line):  # 輸出一行或一列的提示
    remind = []  # 一行或一列的提示記錄
    num = 0  # 提示值
    def fun(line):
        nonlocal remind, num
        flag = 0  # 位移
        if len(line) > 1:
            if line[0] == 0 and line[1] == 1:
                flag += 1
            elif line[0] == line[1] == 0:
                flag += 2
            elif line[0] == 1 and line[1] == 0:
                num += 1
                remind.append(num)
                num = 0
                flag += 2
            elif line[0] == line[1] == 1:
                num += 1
                flag += 1
            fun(line[flag:])
        elif len(line) and line[0]:
            if num:
                remind.append(num + 1)
            else:
                remind.append(1)
    fun(_line)
    return remind

import pygame
import sys
# 參數(shù)設(shè)置 ----------------------------------
blue = (159, 197, 232)  # 被選中方格的顏色
gray = (217, 217, 217)  # 棋盤(pán)網(wǎng)格線(xiàn)顏色
gold = (255, 215, 0)  # 游戲記錄文字顏色
black = (0, 0, 0)
white = (255, 255, 255)
start_x = 240  # 棋盤(pán)左上角位置
start_y = 150
size = 2  # 一行/列的方塊個(gè)數(shù)
square = 320  # 棋盤(pán)邊長(zhǎng)
length = int(square / size)  # 每個(gè)方塊的邊長(zhǎng)
# 游戲初始化 ----------------------------------
pygame.init()
screen = pygame.display.set_mode((780, 520))  # 創(chuàng)建窗口
font = pygame.font.Font(r'./data/msyh.ttf', 20)  # 提示字體
# 主循環(huán) ----------------------------------
while True:
    screen.fill(white)  # 背景填充
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()  # 退出pygame
            sys.exit()  # 安全退出系統(tǒng)
    pygame.display.flip()  # 更新全部顯示

class Item:  # 自定義方塊類(lèi)
    def __init__(self, pos_x, pos_y, leng):
        self.rect = pygame.draw.rect(screen, gray, [pos_x, pos_y, leng, leng], 0)
        self.state = False

def create_chessboard():  # 創(chuàng)建棋盤(pán)
    item_lst = []
    for v in range(size):
        for h in range(size):
            rect = Item(start_x + h*length, start_y + v*length, length)
            item_lst.append(rect)
    return item_lst

def draw_line():  # 繪制網(wǎng)格線(xiàn)
    for n in range(size+1):
        start = (start_x, start_y + n * length)
        end = (start_x + square, start_y + n * length)
        pygame.draw.line(screen, gray, start, end, 2)
    for n in range(size+1):
        start = (start_x + n * length, start_y)
        end = (start_x + n * length, start_y + square)
        pygame.draw.line(screen, gray, start, end, 2)

## 前文參數(shù)、初始化略
......
items = create_chessboard()  # 創(chuàng)建棋盤(pán)
while True:
    screen.fill(white)  # 背景填充
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()  # 退出pygame
            sys.exit()  # 安全退出系統(tǒng)
    draw_line()  # 繪制棋盤(pán)網(wǎng)格線(xiàn)
    pygame.display.flip()  # 更新全部顯示

 if event.type == pygame.MOUSEBUTTONDOWN:  # 鼠標(biāo)點(diǎn)擊事件
        x, y = event.pos

def check_click(item_lst, pos_x, pos_y):  # 更新每個(gè)方塊的點(diǎn)擊狀態(tài)
    for i in item_lst:
        if i.rect.collidepoint(pos_x, pos_y):
            i.state = bool(1 - i.state)

def change_color(item_lst):  # 根據(jù)狀態(tài)改變方塊顏色
    for i in item_lst:
        if i.state:
            pygame.draw.rect(screen, blue, i.rect, 0)
        else:
            pygame.draw.rect(screen, white, i.rect, 0)

def create_answer_array():  # 創(chuàng)建答案矩陣
    lst = [1 if random() > 0.5 else 0 for _ in range(size*size)]
    if list(set(lst))[0] == 0:
        lst[0] = 1
    return lst

def create_answer_array():  # 創(chuàng)建答案矩陣
    lst = [1 if random() > 0.5 else 0 for _ in range(size*size)]
    if list(set(lst))[0] == 0:
        lst[0] = 1
    return lst

## 前文參數(shù)、初始化略
......
answer = create_answer_array()  # 創(chuàng)建答案矩陣
# 主循環(huán) ----------------------------------
while True:
    screen.fill(white)  # 背景填充
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()  # 退出pygame
            sys.exit()  # 安全退出系統(tǒng)
        if event.type == pygame.MOUSEBUTTONDOWN:  # 鼠標(biāo)點(diǎn)擊事件
            x, y = event.pos
            check_click(items, x, y)  # 檢查選中的方格，修改狀態(tài)
            result = get_player_array(items)  # 獲取方格操作矩陣
            print(result)
            if result == answer:
            	print("YOU WIN!")
    change_color(items)  # 根據(jù)方格狀態(tài)修改顏色
    draw_line()  # 繪制棋盤(pán)網(wǎng)格線(xiàn)
    pygame.display.flip()  # 更新全部顯示

def get_w_remind(answer_lst):  # 根據(jù)答案矩陣輸出提示列表
    h_remind = []
    v_remind = []
    h_array = [answer_lst[i: i+size] for i in range(0, len(answer_lst), size)]  # 橫向矩陣
    for h in h_array:
        h_remind.append(get_line_remind(h))
    v_array = list(map(list, zip(*h_array)))  # 縱向矩陣
    for v in v_array:
        v_remind.append(get_line_remind(v))
    return h_remind, v_remind

def show_remind(answer_lst):  # 在棋盤(pán)兩側(cè)對(duì)應(yīng)位置顯示每行/列的提示
    h_remind, v_remind = get_w_remind(answer_lst)
    for i, h in enumerate(h_remind):
        for j, num in enumerate(h[::-1]):
            text = font.render(f"{num}", True, black)
            screen.blit(text, (start_x - 20 * (j + 1), start_y + i * length + length / 2 - 10))
    for i, v in enumerate(v_remind):
        for j, num in enumerate(v[::-1]):
            text = font.render(f"{num}", True, black)
            screen.blit(text, (start_x + i * length + length / 2 - 5, start_y - 30 * (j + 1)))