为何检测列表中整数出现三次及以上的Python代码在特定用例返回错误?
问题原因与修复方案
你的代码错误的核心原因是遍历过程中has_triplet变量会被每个元素的判断结果覆盖,哪怕前面已经找到满足条件的元素,只要最后一个元素不满足,就会返回错误的False。
举个例子,测试用例[1,2,2,2,2,3]:
- 循环到元素2时,
my_list.count(2)等于4,此时has_triplet被设为True - 但循环继续到最后一个元素3时,
my_list.count(3)等于1,has_triplet被覆盖为False - 最终函数返回这个最后一次赋值的False,完全忽略了前面已经存在符合条件的元素
快速修复方案
不需要遍历完所有元素,只要找到第一个出现≥3次的元素,就直接返回True,遍历结束后没有找到再返回False:
def find_if_triplet(my_list): for i in my_list: if my_list.count(i) >= 3: return True return False
更高效的优化方案
因为输入列表是升序排列的,我们可以利用这个特性,只统计连续相同元素的个数,避免每次调用count()带来的重复计算(原方法时间复杂度为O(n²)):
def find_if_triplet(my_list): if len(my_list) < 3: return False current_count = 1 prev_num = my_list[0] for num in my_list[1:]: if num == prev_num: current_count += 1 if current_count >= 3: return True else: current_count = 1 prev_num = num return False
这个方法只需要遍历一次列表,时间复杂度为O(n),在处理长列表时性能提升明显。
内容的提问来源于stack exchange,提问作者thumpy
相关产品推荐
相关产品推荐

