目录
  • 三数之和题目描述
  • 思路
  • Python3代码

三数之和题目描述

给你一个包含 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.00

以上代码有一些思想错误：

遗漏了如果三个数全部大于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 求1-100之间的奇数或者偶数之和的实例
  • Python之list对应元素求和的方法
  • Python基础教程之循环语句（for、while和嵌套循环）