Python单词搜索生成器与单词重叠

import random

def is_valid_placement(grid, word, row, col, direction):
    """检查单词是否可以放置在指定位置和方向"""
    for i, char in enumerate(word):
        r, c = row + i * direction[0], col + i * direction[1]
        if not (0 <= r < len(grid) and 0 <= c < len(grid[0])) or grid[r][c] != ' ' and grid[r][c] != char:
            return False
    return True

def place_word(grid, word, row, col, direction):
    """在网格中放置单词"""
    for i, char in enumerate(word):
        r, c = row + i * direction[0], col + i * direction[1]
        grid[r][c] = char

def generate_word_search(words, grid_size=(10, 10)):
    """生成单词搜索"""
    grid = [[' '] * grid_size[1] for _ in range(grid_size[0])]
    directions = [(0, 1), (1, 0), (0, -1), (-1, 0), (1, 1), (-1, -1), (1, -1), (-1, 1)]
    
    for word in words:
        placed = False
        while not placed:
            row, col = random.randint(0, grid_size[0]-1), random.randint(0, grid_size[1]-1)
            direction = random.choice(directions)
            if is_valid_placement(grid, word, row, col, direction):
                place_word(grid, word, row, col, direction)
                placed = True
    
    return grid

# 示例使用
words = ["PYTHON", "SEARCH", "ALGORITHM", "OVERLAP"]
word_search = generate_word_search(words)
for row in word_search:
    print(''.join(row))