第六届字节跳动青训营第一课 | 豆包MarsCode AI 刷题

188 阅读3分钟

数字分组求偶数和

原题网址:www.marscode.cn/practice/dn…

解题思路

解题的核心在于理解如何从多个数字组中选取一个数字来形成新的数字,并检查这些新数字的各位数之和是否为偶数。以下是详细的解题步骤:

  1. 输入解析:首先,需要将输入的每个数字字符串转换为单个数字的列表。
  2. 生成组合:使用itertools.product生成所有可能的组合,其中每个组合是从每个数字组中选取一个数字。
  3. 计算和检查:对于每个生成的组合,计算其数字之和,并检查该和是否为偶数。
  4. 计数:统计所有使得数字之和为偶数的组合的数量,并返回这个计数。

题目图解

  1. 转换
  • 123 -> [1, 2, 3]
  • 456 -> [4, 5, 6]
  • 789 -> [7, 8, 9]
  1. 生成组合
  • 第一组可能的组合: [1, 4, 7][1, 4, 8][1, 4, 9], …
  • 以此类推,直到 [3, 6, 9]
  1. 计算、检查
  • [1, 4, 7] -> 和为 12 (偶数)
  • [1, 4, 8] -> 和为 13 (奇数)
  • [3, 6, 9] -> 和为 18 (偶数)
  1. 输出
  • 如果三个数的和满足条件(是偶数)那就count++,最后返回count的值

代码详解部分

def solution(numbers):
# 导入函数product,该函数可以用来生成多个列表中所有可能的组合
    from itertools import product
    # 将每个数字组转换为整数列表
    groups = [list(map(int, list(str(num)))) for num in numbers]
    # 生成所有可能的组合,每个组合从每个数字组中取一个数字
    all_combinations = product(*groups)
    # 计算各位数字之和为偶数的组合数量
    count = sum(1 for combo in all_combinations if sum(combo) % 2 == 0)
    return count
if __name__ == "__main__":
    # 测试样例,并打印出结果
    print(solution([123, 456, 789])) 
    print(solution([123456789]))       
    print(solution([14329, 7568])) 

知识总结

  1. 列表推导式:用于将字符串数字转换为单个数字列表,这是一种简洁且高效的方式。
  2. 解包操作符(*) :在调用itertools.product时,解包操作符用于将多个列表作为参数传递。
  3. itertools.product:这个函数用于生成输入列表的笛卡尔积,即所有可能的组合。
  4. 条件表达式:在判断数字之和是否为偶数时,使用了条件表达式 sum(combo) % 2 0,这是一种简洁的条件检查方式。
  5. 迭代与枚举:通过迭代所有可能的组合,并枚举每个组合来计算和检查,这是解决此类问题的常见方法。
  6. 模块化编程:将代码组织成函数,提高了代码的可读性和可重用性。
  7. 主函数与脚本运行:理解 if name “main”: 的用法,确保代码块只在脚本直接运行时执行,这是Python脚本编写的一个常见实践。

建议

  • 在解决组合问题时,建议先理解问题的本质,然后熟悉并利用Python标准库中的函数,如itertools.product,以简化问题求解过程。
  • 刷题时,先认真看下题,自己尝试去做一下,没有思路,用marscode给出思路,一点点修改,接近正确答案,写完一道题之后,尽量地去写笔记,这样能够更好的使自己巩固知识点。
  • 同时,保持一个良好的刷题习惯,我们经常说“千里之行,始于足下”,做算法题,不是一朝一夕的成果,而是需要长期去坚持的,所以希望大家一起加油,一起坚持,为了明天更好的自己!