day7 代码随想录 | 15 三数之和 18 四数之和 454 四数相加

最新推荐文章于 2024-10-04 20:45:38 发布

Zison_Chen

最新推荐文章于 2024-10-04 20:45:38 发布

阅读量674

点赞数 27

文章标签：算法数据结构 python leetcode

本文链接：https://blog.csdn.net/Zison_Chen/article/details/140633817

版权

三数之和

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

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

这个题目题目要读明白。首先，你取得数的索引是不能相同的。其次，即使你是取了三个数索引

不相同，仍然可能包含重复的三元组。什么意思？比如数组是【0，0，0，0】按照不同索引，你可以得到几个[0,0,0]的结果，但是你需要去重。

由于之前做过两数之和，0就是我们所谓的target，那么是否可以采用类似的思路，用map取去映射呢。理论上是可以的，但是去重操作就非常麻烦。当你用key,value分别存储值和索引时，对于重复值的值你会覆盖索引，同时，你需要再次操作去去除最后所获得元组的重。很麻烦。

因此，看似是一个map找值的问题，其实并不适合这种解法。

双指针

但我们可以采用双指针的思路，参考代码随想录。

我们将数组排序，初始化i指针，指向的值为a，遍历数组，left指针指向i+1,指向的值为b, right指针指向数组的尾部,指向的值为c。

这个我们就获得了了三个数值。由于排序的数组，如果加和的总数大于0，right向左移动；小于0，那么left向右边移动。直到相遇。

关键在于去重。这里我们已经实现了索引都不一样，但是对于a, b, c我们没有去重。这里的去重是指，我们还是举例说明，比如整个的数组为[-1, -1, -1, 2]。a刚开始为-1(索引为0)，此时left为-1(索引为1)， right = -2.此时就开始遍历，会得到结果-1(索引0)，-1(索引1），2(索引3), -1(索引0)，-1(索引2），2(索引3)[这里的重复是left指针的问题]。当i继续向前移动，你一会会得到结果-1(索引0)，-1(索引2），2(索引3)。这个索引2和索引3的结果。因为你之前的遍历已经包含了这种可能了。因此你需要去重a, 去重a的方式就是当a[i] == a[i-1] 的时候，直接进入小一轮循环，索引i直接+1。为什么不是a[i]=a[i+1]。思路上是当i=0时，a[0]=a[1](上面这个例子)，直接会跳过这一轮，但是这一轮不应该跳过，因为这一轮遍历的范围最大。会跳过正确答案。另外，a[i]=a[i+1],会导致索引越界，当i=n-1(数组长度为n)。

对于b,c，就是left，right指针的去重。就很简单，分别访问和比较它的左边和右边的元素，如果相同，指针移动。

def threeSum(nums: List[int]) -> List[List[int]]:
    res = []
    nums.sort()
    
    n = len(nums)

    for i in range(n):
        if nums[i] > 0:
            return res
        
        if (i > 0 and nums[i] == nums[i-1]):
            continue

        left = i + 1
        right = n - 1

        while right > left:
            a = nums[i]
            b = nums[left]
            c = nums[right]
            sum_ = a + b + c
            
            if sum_ > 0:
                left += 1

            elif sum_ < 0:
                right -= 1

            else:
                res.append([a, b, c])
                
                while (right > left and nums[left] == nums[left+1]): left += 1
                while (right > left and nums[right] == nums[right-1]): right -= 1

                right -= 1
                left += 1

      return res

四数之和

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

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

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

示例 1：

输入：nums = [1,0,-1,0,-2,2], target = 0
输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

其实和三数之和完全一个逻辑。区别在于多一层循环，多添加一个指针。思路一样，

具体可查看代码随想录。

def fourSum(nums: List[int], target: int) -> List[List[int]]:
    res = []
    nums.sort()

    n = len(nums)

    for i in range(n):
        if (i > 0 and nums[i] == nums[i-1]):
            continue

        for j in range(i+1, n):
            if (j > i+1 and nums[j] == nums[j-1]):
                continue

            left = j + 1
            right = n-1

            while right > left:
                a = nums[i]
                b = nums[j]
                c = nums[left]
                d = nums[right]

                if (a + b + c + d) > target:
                    right -= 1

                else(a+b+c+d) < target:
                    left += 1

                else:
                    res.append([a, b, c, d])

                    while right > left and nums[left] == nums[left+1]: left += 1
                    while right > left and nums[right] == nums[right-1]: right -= 1

                    left += 1
                    right -= 1

       return res

四数相加

给你四个整数数组 nums1、nums2、nums3 和 nums4 ，数组长度都是 n ，请你计算有多少个元组 (i, j, k, l) 能满足：

0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

输入：nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
输出：2

这里题目一定要理解题意，与之前的四数之和的区别。

这里是4个独立数组，每个数组里面取一个数来加和，看最终是否为0，不是从一个数组里去看！

还有就是这个不要求返回具体的组合，只需要知道多少个。这里其实就简单非常多。

我们可以分为两组，num1 + num2 构建一个字典，key是组合的值，value是数量。然后用num3和num4的组合去匹配，然后相加这个数量就可以了！这里只需要索引不重复，num相加不是自己加自己就可以了。

def fourSumCount(nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
    dic = {}
    count = 0
    for num1 in nums1:
        for num2 in nums2:
            s = num1 + num2
            dic[s] = dic.get(s, 0) + 1

    for n3 in nums3:
        for n4 in nums4:
            s = -n3-n4
            if dic.get(s, 0) != 0:
                count += dic[s]

    return count