字母随机数字分配与单词平方和计算代码问题及实现需求

原代码存在的错误 #

• 字典word_map包含重复键（M和E各出现两次），Python字典会自动覆盖重复键的最后一个值，导致键值对不完整且不符合随机分配的需求。
• 循环逻辑错误：word.val.append(word_map[word])中，word是字符串（如'MY'），并非带有val属性的对象；且错误地将整个单词作为字典键，正确逻辑应遍历单词中的每个字母并累加对应数字的和。
• 未实现核心需求：没有编写随机分配数字并确保每个单词和为平方数的逻辑，仅硬编码了字典值。

修正后的代码 #

import random

def is_square(n):
    # 判断一个数是否为完全平方数
    if n < 0:
        return False
    root = int(n ** 0.5)
    return root * root == n

words = ['MY', 'NAME', 'IS', 'E']
# 提取所有唯一字母
unique_letters = list(set(''.join(words)))
letter_map = {}

# 循环尝试随机分配数字，直到满足所有单词和为平方数
while True:
    # 为每个字母随机分配1-20之间的整数（可调整范围）
    letter_map = {letter: random.randint(1, 20) for letter in unique_letters}
    # 检查所有单词的和是否都是平方数
    all_valid = True
    word_sums = {}
    for word in words:
        total = sum(letter_map[char] for char in word)
        word_sums[word] = total
        if not is_square(total):
            all_valid = False
            break
    if all_valid:
        break

# 按指定格式输出结果
output = ", ".join([f"{word}: {word_sums[word]}" for word in words])
print(output)
# 同时输出字母对应的数字映射
print("字母对应数字：", letter_map)

代码说明 #

• is_square函数：通过计算平方根取整后平方是否等于原数，判断一个数是否为完全平方数。
• 提取唯一字母：将所有单词拼接成字符串后去重，得到需要分配数字的全部字母。
• 随机分配逻辑：循环为字母随机分配1-20的整数（可自行调整数字范围），直到所有单词的字母和均为平方数。
• 结果输出：按MY: X, NAME: Y的格式拼接输出每个单词的和，同时打印字母对应的数字映射。

内容的提问来源于stack exchange，提问作者aurora