Python糖果游戏

import numpy as np

N = 5  # 棋盘大小为5x5
board = np.random.randint(1, 4, size=(N, N))  # 随机生成1-3之间的整数作为糖果颜色

print(board)

[[1 2 3 3 2]
 [2 1 1 2 2]
 [1 3 2 1 2]
 [3 3 2 2 1]
 [3 2 1 3 1]]

def eliminate(board):
    def dfs(x, y, color, visited):
        if x < 0 or x >= N or y < 0 or y >= N or visited[x][y] or board[x][y] != color:
            return 0
        visited[x][y] = True
        count = 1
        count += dfs(x + 1, y, color, visited)
        count += dfs(x - 1, y, color, visited)
        count += dfs(x, y + 1, color, visited)
        count += dfs(x, y - 1, color, visited)
        return count

    visited = np.full((N, N), False)
    score = 0
    for i in range(N):
        for j in range(N):
            if not visited[i][j]:
                color = board[i][j]
                count = dfs(i, j, color, visited)
                if count >= 3:
                    board[visited] = 0
                    score += count
    return score

print(eliminate(board))
print(board)

8
[[1 0 0 0 0]
 [2 0 0 0 0]
 [1 0 0 1 0]
 [3 3 0 2 1]
 [3 2 0 3 1]]

def swap(board, x1, y1, x2, y2):
    board[x1][y1], board[x2][y2] = board[x2][y2], board[x1][y1]

swap(board, 0, 0, 0, 1)
print(board)

[[2 1 0 0 0]
 [2 0 0 0 0]
 [1 0 0 1 0]
 [3 3 0 2 1]
 [3 2 0 3 1]]