Python中3Sum的使用

假设我们有一个数字数组。它存储n个整数，数组中有元素a，b，c，这样a+b+c=0。找到满足情况的数组中的所有唯一三元组。因此，如果数组类似于[-1,0,1,2，-1，-4]，则结果将为[[-1，1，0]，[-1，-1，2]]

为了解决这个问题，我们将遵循以下步骤-

对数组num进行排序，并定义数组res

对于0到num的长度范围内的i–3

sum：=nums[i]，nums[l]和nums[r]的总和

如果sum<0，则l：=l+1，否则，当sum>0时，则r：=r–1

否则将nums[i]，nums[l]，nums[r]插入res数组

而l<num的长度–1和nums[l]=nums[l+1]

而r>0且nums[r]=nums[r-1]

将l增加1并将r减少1

使我增加1

将r减1

如果i>0并且nums[i]=nums[i-1]，则跳过下一部分并继续

l：=i+1和r：=nums的长度–1

当l<r

返回资源

示例(Python)

让我们看下面的实现以更好地理解-

class Solution(object): def threeSum(self, nums): nums.sort() result = [] for i in range(len(nums)-2): if i> 0 and nums[i] == nums[i-1]: continue l = i+1 r = len(nums)-1 while(l<r): sum = nums[i] + nums[l] + nums[r] if sum<0: l+=1 elif sum >0: r-=1 else: result.append([nums[i],nums[l],nums[r]]) while l<len(nums)-1 and nums[l] == nums[l + 1] : l += 1 while r>0 and nums[r] == nums[r - 1]: r -= 1 l+=1 r-=1 return result ob1 = Solution() print(ob1.threeSum([-1,0,1,2,-1,-4]))

输入项

[-1,0,1,2,-1,-4]

输出结果

[[-1,-1,2],[-1,0,1]]