AI 刷题 4. 数字分组求偶数和 题解 | 豆包MarsCode AI刷题

题目描述

问题：小M面对一组从 1 到 9 的数字，这些数字被分成多个小组，并从每个小组中选择一个数字组成一个新的数。目标是使得这个新数的各位数字之和为偶数。任务是计算出有多少种不同的分组和选择方法可以达到这一目标。

输入：一个由多个整数字符串组成的列表 numbers，每个字符串可以视为一个数字组。

输出：计算出有多少种不同的分组和选择方法可以达到这一目标。

解题思路

1. 拆分数字组：

   • 将每个数字组拆分为单个数字。例如，123 可以拆分为 [1, 2, 3]。

2. 组合选择：

   • 使用组合的方式，从每个数字组中选择一个数字，组成一个新的数。

3. 判断偶数和：

   • 计算新数的各位数字之和，判断其是否为偶数。

4. 计数：

   • 统计所有符合条件的组合数量。

数据结构与算法

• 数据结构：

  • 使用列表来存储每个数字组的拆分结果。
  • 使用嵌套循环来生成所有可能的组合。

• 算法步骤：

  1. 将每个数字组拆分为单个数字。
  2. 使用嵌套循环生成所有可能的组合。
  3. 计算每个组合的各位数字之和。
  4. 判断和是否为偶数，并统计符合条件的组合数量。

代码实现：

import itertools

def solution(numbers):
    # 将每个数字组拆分为单个数字
    split_numbers = [list(map(int, str(num))) for num in numbers]
    
    # 初始化计数器
    count = 0
    
    # 使用嵌套循环生成所有可能的组合
    for combination in itertools.product(*split_numbers):
        # 计算组合的各位数字之和
        total_sum = sum(combination)
        
        # 判断和是否为偶数
        if total_sum % 2 == 0:
            count += 1
    
    return count

if __name__ == "__main__":
    # 你可以添加更多测试用例
    print(solution([123, 456, 789]) == 14)
    print(solution([123456789]) == 4)
    print(solution([14329, 7568]) == 10)

核心要点：

total_sum = sum(combination)   **计算和判断**
if total_sum % 2 == 0:
    count += 1