Python三数之和的实现方式
更新时间：2022年05月11日 10:30:19 作者：每天收获一点点
这篇文章主要介绍了Python三数之和的实现方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
三数之和题目描述
给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，
使得 a + b + c = 0 ？请你找出所有满足条件且不重复的三元组。
答案中不允许包含重复的三元组。
示例：
给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[ [-1, 0, 1], [-1, -1, 2] ]
思路
1. 首先将数组排序，可以利用Python内置函数，也可以利用另外定义排序算法。
2. 应用双指针算法。固定第一个数,索引为i，遍历整个数组，第一个数也是三个数中最小的数，然后在该数右面设置左右两个指针l和r,l=i+1，r=len(nums)-1,
3. 判断这三个索引指向的元素和与0的大小关系。
和>0,右指针左移一位；和<0，左指针右移一位。
由于要避免重复的三元组，所以移动左右指针的时候要跳过相邻的所有相等的nums[i]。
Python3代码
#导入计算时间的包，调用系统时间
from time import *
#初始时间
t1 = time()
def threeSum(nums):
nums.sort()
n = len(nums)
res = []
for i in range(n):
'''如果相邻的两个数相等，跳过,避免重复'''
if i > 0 and nums[i] == nums[i-1]:
continue
l, r = i+1, n-1
while l < r:
if nums[i] + nums[l] + nums[r]>0:
r -= 1
while nums[r+1] == nums[r]:
r -= 1
elif nums[i] + nums[l] + nums[r]<0:
l += 1
while nums[l-1] == nums[l]:
l += 1
else:
res.append([nums[i],nums[l],nums[r]])
l += 1
r -= 1
while nums[l] == nums[l - 1]: l += 1
while nums[r] == nums[r + 1]: r -= 1
return res
if __name__ == '__main__':
nums = [-1,0,1,2,-1,-4]
print(threeSum(nums))
#结束时间
t2 = time()
#运行时间
run_time = t2 - t1
print(run_time)
运行结果：
[[-1, -1, 2], [-1, 0, 1]]
#运行时间
0.0010113716125488281
以上代码有一些思想错误：
遗漏了如果三个数全部大于0，则退出循环，因为没有满足条件的结果。
没有严格判断每一次的l<r的条件。
修正后的代码：
from time import *
t1 = time()
def threeSum(nums):
nums.sort()
n = len(nums)
res = []
for i in range(n-2):
if nums[i] > 0:break
'''如果相邻的两个数相等，跳过,避免重复'''
if i > 0 and nums[i] == nums[i-1]:
continue
l, r = i+1, n-1
while l < r:
if nums[i] + nums[l] + nums[r]>0:
r -= 1
while l < r and nums[r-1] == nums[r]:
r -= 1
elif nums[i] + nums[l] + nums[r]<0:
l += 1
while l < r and nums[l] == nums[l-1]:
l += 1
else:
res.append([nums[i],nums[l],nums[r]])
l += 1
r -= 1
while l < r and nums[l] == nums[l - 1]: l += 1
while l < r and nums[r] == nums[r + 1]: r -= 1
return res
if __name__ == '__main__':
nums = [-2,-3,0,0,-2]
print(threeSum(nums))
t2 = time()
run_time = t2 - t1
print(run_time)
结果：
[]
#时间
0.0
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
Python为Excel单元格设置填充\背景色的操作方法(纯色、渐变、图案)
在使用Excel进行数据处理和分析时,对特定单元格进行背景颜色填充不仅能够提升工作表的视觉吸引力,还能帮助用户快速识别和区分不同类别的数据,本文将通过三个示例详细介绍如何使用Python在Excel中设置不同的单元格背景,需要的朋友可以参考下2024-08-08
今天小编就为大家分享一篇关于numpy数组轴的使用详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
这篇文章主要介绍了Python读取CSV文件的四种方法， 使用 csv 模块，使用 numpy 库，使用pandas库，使用标准库中的 csv 模块这四种方法，文章通过代码示例介绍的非常详细,需要的朋友可以参考下2023-07-07
在本篇文章里小编给大家整理的是一篇关于python dir函数快速掌握用法技巧，有兴趣的朋友们可以学习参考下。2020-12-12
大家好，本篇文章主要讲的是Python制作个性化的词云图实例讲解，感兴趣的同学赶紧来看一看吧，对你有帮助的话记得收藏一下2022-02-02
这篇文章主要介绍了使用Python对文件进行批量改名的方法,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2023-03-03
这篇文章主要介绍了Python Selenium常见的报错问题以及措施,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2024-05-05
浅析python，PyCharm，Anaconda三者之间的关系
Python是一种跨平台的计算机程序设计语言，简单来说，python就是类似于C，Java，C++等，一种编程语言。这篇文章主要介绍了浅析python，PyCharm，Anaconda三者之间的关系,需要的朋友可以参考下2019-11-11
详解Python list和numpy array的存储和读取方法
这篇文章主要介绍了详解Python list和numpy array的存储和读取方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-11-11
这篇文章主要介绍了Python中的SOAP项目及其在web开发中的应用,本文来自于IBM官方网站技术文档,需要的朋友可以参考下2015-04-04
最新评论