Python递归生成n元集合r元素组合 代码错误排查及方法咨询

你的核心思路是对的（组合拆分含首元素/不含首元素），但代码实现有几个明显错误：

• 滥用全局变量y：递归过程中所有层级都往同一个全局列表追加内容，导致最终结果嵌套层级混乱、重复数据多，递归函数本身应该直接返回当前输入对应的组合结果，不需要额外全局变量存储。
• 终止条件错误：你写的len(sol)==1逻辑完全不成立，比如剩余元素长度是2、要求选3个元素的场景，根本凑不出符合要求的组合，应该直接返回空列表，错误的终止条件会返回大量无效结果。
• 列表拼接逻辑错误：[[sol[0]] + combinations(sol[1:], b-1)]的写法不符合预期，combinations(sol[1:], b-1)返回的是多个子组合的列表，你需要给每个子组合的头部都加sol[0]，而不是把sol[0]直接拼在子组合列表的头部，会导致结构完全错位。
• 基础语法错误：最后print语句缺少右括号，运行会直接报语法错误。

首先明确递归函数的定义：combinations(arr, k)的作用是返回数组arr中所有长度为k的组合列表，全程遵循这个定义即可。

def combinations(arr, k):
    # 终止条件1：要选0个元素，只有空列表这一种结果
    if k == 0:
        return [[]]
    # 终止条件2：剩余元素长度小于要选的数量，凑不出来，返回空
    if len(arr) < k:
        return []
    # 终止条件3：剩余元素长度刚好等于要选的数量，直接返回当前数组作为唯一结果
    if len(arr) == k:
        return [arr.copy()]
    # 情况1：包含第一个元素，从后面的元素选k-1个，每个结果前面加首元素
    include_first = [[arr[0]] + sub for sub in combinations(arr[1:], k-1)]
    # 情况2：不包含第一个元素，从后面的元素选k个
    exclude_first = combinations(arr[1:], k)
    # 合并两种情况返回
    return include_first + exclude_first

sol = [1,2,3,4,5]
b = 3
print(combinations(sol, b))

运行后输出和你预期的结果完全一致：[[1, 2, 3], [1, 2, 4], [1, 2, 5], [1, 3, 4], [1, 3, 5], [1, 4, 5], [2, 3, 4], [2, 3, 5], [2, 4, 5], [3, 4, 5]]

• 先从小案例手动推导递归过程：比如先尝试写从[1,2,3]选2个元素的组合，手写每一层递归的输入、输出，搞清楚每一步的结果结构，不要一上来就写复杂的大输入。
• 初期不要用全局变量存结果：优先让递归函数本身返回对应层级的结果，出现问题时可以逐层打印返回值，很容易定位结构错误。
• 先明确递归定义再写代码：动笔前先写清楚函数的输入、返回值含义，所有递归调用都要符合这个定义，不要写着写着就偏离初始逻辑。

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