本文主要是介绍代码随想录 -- 哈希表 -- 三数之和,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
15. 三数之和 - 力扣(LeetCode)
思路:
本题适合采用双指针法。
1. 先将数组 nums 升序排序
2. 初始状态:i = 0 ,指针 left = i+1 ,指针 right = len(nums)-1 (left 要 < right)
3. 如果 nums[i] > 0,则不存在满足条件的三元组;
如果 nums[i] + nums[left] + nums[right] > 0,right 指针向前移动;
如果 nums[i] + nums[left] + nums[right] < 0,left 指针向后移动;
如果 nums[i] + nums[left] + nums[right] == 0,找到符合条件的三元组,加入res列表。
4. 去重:
对于 i:当 i > 0 时,如果 nums[i] == nums[i-1],i += 1;
对于 left:如果 nums[left] == nums[left-1],left += 1;
对于 right:如果 nums[right] == nums[right+1],right -= 1.
class Solution(object):def threeSum(self, nums):nums.sort()res=[]for i in range(len(nums)):if nums[i]>0:return resif i>0 and nums[i]==nums[i-1]:continueleft,right=i+1,len(nums)-1while left<right:sum = nums[i]+nums[left]+nums[right]if sum>0:right-=1elif sum<0:left+=1else:res.append([nums[i],nums[left],nums[right]])while left<right and nums[right]==nums[right-1]:right-=1while left<right and nums[left]==nums[left+1]:left+=1left+=1right-=1return res
这篇关于代码随想录 -- 哈希表 -- 三数之和的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!