18. 四数之和

引流:五点钟科技-CSDN博客 

题目:

给你一个由 n 个整数组成的数组 nums ,和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] (若两个四元组元素一一对应,则认为两个四元组重复):

  • 0 <= a, b, c, d < n
  • abc 和 d 互不相同
  • nums[a] + nums[b] + nums[c] + nums[d] == target

你可以按 任意顺序 返回答案 。

18. 四数之和_第1张图片

【解题思路】 

本题虽然是四数之和,但和我们之前做过的所有x数之和都是换汤不换药,思路是一样的,只不过这里可能需要多一个循环,即4个指针共同来完成此题。

    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        if len(nums) < 4:
            return []
        nums.sort()
        n = len(nums) - 1
        res = []
        for i in range(n-2):
            if i and nums[i] == nums[i-1]:
                continue
            for j in range(i+1, n-1):
                if j > i+1 and nums[j] == nums[j-1]:
                    continue
                left = j + 1
                right = n
                while left < right:
                    if left > j + 1 and nums[left]==nums[left-1]:
                        left += 1
                        # 这里需要及时continue掉,否则如果left+1后仍然满足nums[left]==nums[left-1],
                        # 继续往下执行,仍然会出现重复的答案,例如当nums=[2, 2, 2, 2, 2],下面的right同理。
                        continue
                    if right < n and nums[right] == nums[right+1]:
                        right -= 1
                        continue
                    # 一定要先判断重复项,再在此处判断x和target的关系
                    x = nums[i]+nums[j]+nums[left]+nums[right]
                    if x == target:
                        res.append([nums[i], nums[j], nums[left], nums[right]])
                        left += 1
                        right -= 1
                    elif x < target:
                        left += 1
                    else:
                        right -= 1
        return res

18. 四数之和_第2张图片

你可能感兴趣的:(一起来刷题,#,和的问题,算法,数据结构,python,leetcode,刷题)