如何实现无切片递归函数判断列表/子列表和等于正整数k?
递归实现子列表和匹配问题
你这段代码的问题还挺明显的,我帮你拆解下:
- 首先你用了
itertools.combinations生成子列表,但这完全偏离了题目要求的递归思路,而且这种方式效率不高,还容易触发错误 - 循环里的
n-i-1会随着i增大变成负数,比如当i >= n时,这个值就小于0了——这就是你看到ValueError: r must be non-negative的原因,combinations的第二个参数不能是负数 - 递归调用的时候你打错了函数名(应该是
sum_rec不是s_rec),而且就算调用对了,你也没处理它的返回值——哪怕某个子列表满足和等于k,你的代码也不会把这个结果传递回去,最后只会走到else返回False - 你还漏掉了单个元素的判断:当列表只剩一个元素时,你没检查它的值是不是等于k
接下来我给你两个符合要求的递归实现,完全不用切片,纯递归逻辑:
方法一:通过弹出/插入元素避免切片
这个方法会临时修改列表,但递归结束后会恢复原列表,保证外部使用不受影响:
def sum_rec(L, k): # 基线条件1:如果目标和已经减到0,说明找到符合条件的子列表 if k == 0: return True # 基线条件2:列表为空,没找到符合条件的子列表 if not L: return False # 取出第一个元素,避免使用切片 current = L.pop(0) # 两种递归分支:包含当前元素,或者不包含 include = sum_rec(L, k - current) exclude = sum_rec(L, k) # 把元素插回去,恢复原列表 L.insert(0, current) # 只要其中一种情况成立,就返回True return include or exclude
方法二:用索引跟踪处理位置(更高效)
这个版本不需要修改原列表,通过索引来标记当前处理到哪个元素,逻辑更清晰:
def sum_rec(L, k): # 定义辅助递归函数,用index跟踪当前处理的位置 def helper(index, target): # 基线条件:找到符合条件的子列表 if target == 0: return True # 基线条件:遍历完所有元素仍未找到 if index >= len(L): return False # 分支1:包含当前索引的元素,目标和减去该元素值 include = helper(index + 1, target - L[index]) # 分支2:不包含当前索引的元素,目标和不变 exclude = helper(index + 1, target) return include or exclude return helper(0, k)
测试示例
sum_rec([1,2,3], 5)→ 返回True(2+3=5)sum_rec([1,2,3], 7)→ 返回False(所有元素和为6)sum_rec([5],5)→ 返回Truesum_rec([],0)→ 返回True(空列表的和为0)
内容的提问来源于stack exchange,提问作者roman asa
相关产品推荐
相关产品推荐

