LeetCode 三数之和(双指针法)(算法练习)

    科技2024-01-30  110

    给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。

    注意:答案中不可以包含重复的三元组。

    示例:

    给定数组 nums = [-1, 0, 1, 2, -1, -4],

    满足要求的三元组集合为: [ [-1, 0, 1], [-1, -1, 2] ]

    来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/3sum 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

    nums = [-2,0,0,2,2] rs = [] # 对数列进行递增排序 nums.sort() # nums = [-1,-1,-1,0,1,2] # print(nums) # 设置最小的第一个 指针 k for k in range(len(nums) - 2): # 设置l,r指针 r = len(nums) - 1 # 指向最大 K 变化一次再次指向最后 l = k + 1 # 不断指向下一个变换的 # 最小的指针指向大于1 与指向相同的结果时跳过 # print('k: %d,l: %d,r: %d' % (k, l, r)) # print('nums[k]: %d,nums[l]: %d,nums[r]: %d' % (nums[k], nums[l], nums[r])) if nums[k] > 0: break if (k > 0) and (nums[k] == nums[k - 1]): continue while (k + 1) <= l < r: # 处理相同的指针元素 if (l > k + 1) and (nums[l] == nums[l - 1]): l += 1 continue if (r < len(nums) - 1) and (nums[r] == nums[r + 1]): r -= 1 continue s = nums[k] + nums[l] + nums[r] # print(s) # s == 0 说明结果符合 将索引指向元素记录 if s == 0: rs.append([nums[k], nums[l], nums[r]]) l += 1 r -= 1 # s < 0 的情况 说明 l 指针需要向后移一位 elif s < 0: l += 1 # s>0 的情况说明 r 指针需要往前移一位 else: r -= 1 print(rs)
    Processed: 0.010, SQL: 8