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]]